o
    y;j                     @   s  d dl Z d dlZd dlZd dlZd dlZd dlZe je je	Z
dZdZe je
dZe je
dZe Zejddd ejd	d
d e Ze je de   Ze je de   Ze jeddd e jeddd e j Zeed< eed< eed< ded< eed< e jeje j edd ed< ej dd>Z!e j"ej#Z$e je!e$Z%eej#e% ejedededede!dej&gZ'e(de' e(d ej)e'ded  W d   n1 sw   Y  e(d! e(d"ej& dS )#    Nlip0z)checkpoints/exp-schp-201908261155-lip.pthzsimple_extractor.py
image_pathu   입력 이미지 경로)help
output_diru   출력 폴더schp_cache_schp_torch_extensions_i  T)modeexist_okHOMEXDG_CACHE_HOMETORCH_EXTENSIONS_DIRz8.9TORCH_CUDA_ARCH_LISTCUDA_VISIBLE_DEVICESPATH schp_inputs_)prefixz	--datasetz--gpuz--model-restorez--input-dirz--output-dirzcmd: )checkenvu   SCHP 실행 완료u   결과 폴더:)*os
subprocessargparseshutilsystempfilepathdirnameabspath__file__
SCRIPT_DIRDATASETGPU_IDjoin
CHECKPOINTSIMPLE_EXTRACTORArgumentParserparseradd_argument
parse_argsargs
gettempdirgetuid
cache_hometorch_extensions_dirmakedirsenvironcopyr   
executablepathsepgetTemporaryDirectory	input_dirbasenamer   
image_nametarget_pathr   cmdprintrun r=   r=   htdocs/OOTD3/SCHP/schp_test.py<module>   sX    
$

