o JO<||dd|d?d@d|>O<||dd|d?d@d|>O<|d7}|dL}|s.q|S)z Returns the color map for visualizing the segmentation mask. Args: num_cls: Number of classes Returns: The color map rr#)range)num_clsnpalettejlabir9r9r: get_paletteFs"(((rFcCs,t|tj}|D]}d|||k<q |S)u7특정 클래스만 흰색(mask)으로 만드는 함수)np zeros_likeastypeuint8)parsing_result class_idsmaskcidr9r9r: create_mask^srPc Cst|dk\}}t|dkst|dkr|S|j\}}t|td}t|td}t|td|}t|td|} ||| ||fS)uB투명하지 않은 영역만 남기도록 이미지 여백 제거rr=)rHwherelenshapemaxmin CROP_PADDING) imagealphaysxsheightwidthx1y1x2y2r9r9r: crop_to_alphafs racCsPt|||g}t|dk|d}t|tj}||dddddf<t||S)u.마스크를 적용하여 의류 부분 추출rGrNr<)cv2mergerHrQcvtColorCOLOR_RGB2RGBAra)original_imagerNmask_3ch extractedrgbar9r9r: extract_partvs rjc. CsHt}|jdkr tddd|jdD}t|dksJ|jtjd<tj s.tdt d }t |j d }t |j d }t |j d }t d ||t d|tjd|dd}tj|j|dd}ddlm}|} |D]\} } | dd} | | | <qt|| |||tttjgdgddg} t|j|| d}t|}tj !|j"rddl#}|$|j"t%|j"t&|}dgdgdgdgdgdgdgdgd }t'7t(t)|D]&\}}|\}}|d!d}|d"*d}|d#*d}|d$*d}|d%*d}|||}tj+j,|d&d'd(}||dd)d-d}|.}|/dd*d}t0|j12*|||||d+}t3j4|d*d,} |j5rntj 6|j"|dd-d.}!t37|!|t d/|tj 6|j|}"t89|"}#t8:|#t8j;}#t8<|#| j=d| j=df}$t d0|d1|D]l\}%}&t>| |&}'|'?dkr|&d}(|dddd|(f})|)|'dk@}*ddt3A|* }+t |%d2d3|+d4d5d6tB|$|'},tj 6|j"|dd-d7|%d8}-t8C|-t8:|,t8jDqt |%d2d9qqWddS1swYdS):NNoneuPGPU 전용 실행만 허용됩니다. --gpu 값은 None이 될 수 없습니다.cSsg|]}t|qSr9)int).0rEr9r9r: szmain..,r=CUDA_VISIBLE_DEVICESuMCUDA GPU를 사용할 수 없습니다. GPU 전용 실행만 허용됩니다.zcuda:0rrrz(Evaluating total class number {} with {}z Using device: resnet101)r pretrained) map_location state_dictr) OrderedDictr#)gCl?gv/?g ףp= ?)g?gy&1?gZd;O?)meanstd)rootr transform r)hat upper_clothesdresscoatpants jumpsuitsscarfskirtnamecenterscaler\r[bilinearT)sizemode align_cornersr>)r)axisz.npyu의류별 추출 중: z === u 의류 확률 ===15z : dz6.2f%_z.pngu : 감지 안됨)Er;gpu RuntimeErrorsplitrRosenvirontorchcuda is_availabledevicedataset_settingsdatasetprintformatnetworks init_modelload model_restore collectionsruitemsload_state_dicttoeval transformsComposeToTensor Normalizer input_dirrpathexists output_dirshutilrmtreemakedirsrFno_grad enumeraternumpynnUpsample unsqueezesqueezepermuterdatacpurHargmaxlogitsjoinsaverbimreadrd COLOR_BGR2RGBresizerSrPsumrvexprjimwriteCOLOR_RGBA2BGRA).argsgpusrrrrmodelrtrunew_state_dictkvrryr dataloaderrrBclothing_classesidxbatchrWmetaimg_namecswhoutputupsampleupsample_output logits_resultrLlogits_result_pathoriginal_image_pathrforiginal_image_resized clothing_namerM clothing_maskclass_id class_scores average_score softmax_prob clothing_imgclothing_output_pathr9r9r:mains                  $ 77r__main__)__doc__rrbrr3rrHPILrrtorch.utils.datartorchvision.transformsrrutils.transformsr!datasets.simple_extractor_datasetrrVrr;rFrPrarjr__name__r9r9r9r:sH        v