o +s  z"get_image_files..)sortedrlistdir) input_dirrrrget_image_files*sr;c Cst}tj|j}tj|j}tj|std|t|}|s+t d|tj |ddt j t dtdtdtd|d |g }td |td t||D]}td |qOtd |td|tdtj|dttdtdtd |dS)Nu입력 폴더가 없습니다: u,입력 폴더에 이미지가 없습니다: T)rz --datasetz--gpuz--model-restorerr u입력 폴더:u이미지 수:z -u결과 폴더:zcmd:r)checkr0cwduSCHP 실행 완료)rrr abspathr: output_dirisdirFileNotFoundErrorr; RuntimeErrorr%r*r+SIMPLE_EXTRACTORDATASETr( CHECKPOINTprintlen subprocessrunr1SCHP_DIR)argsr:r? image_filescmd image_filerrrmain2s6     rO__main__)r rrHr*r"r r)r>__file__ SCRIPT_DIRr!rJrDr(rErCrrr4rr1r;rO__name__rrrrs*&