개요

CycleGan, CycleGan을 직접 PyTorch로 돌려보기 위한 스크립트를 돌려보고 정리해둔다.

사용방법

Torch 버전

$ export CUDNN_PATH="/home/lks21c/cudnn_50/lib64/libcudnn.so.5"
$ DATA_ROOT=./datasets/ae_photos name=style_cezanne_pretrained model=one_direction_test phase=test loadSize=256 fineSize=256 resize_or_crop="scale_width" th test.lua

PyTorch 버전

학습 수행은 아래와 같다.

$ python -m visdom.server
$ python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

테스트는 아래와 같다.

$ python test.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

결과는 ./results/maps_cyclegan/latest_test/index.html에서 확인 가능하다.