o
    
z;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
e je
d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SCHP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joinSCHP_DIRDATASETGPU_ID
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?   schp_test.py<module>   sZ    
$

