데이터셋 다운로드

$ bash ./datasets/download_cyclegan_dataset.sh maps

학습

#!./scripts/train_cyclegan.sh
python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan    

테스트

#!./scripts/test_cyclegan.sh
python test.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

결과는 ./results/maps_cyclegan/latest_test/index.html에서 확인

모델 로드하기

$ bash ./scripts/download_cyclegan_model.sh horse2zebra

모델 리스트

  • apple2orange
  • orange2apple
  • summer2winter_yosemite
  • winter2summer_yosemite
  • horse2zebra
  • zebra2horse
  • monet2photo
  • style_monet
  • style_cezanne
  • style_ukiyoe
  • style_vangogh
  • sat2map
  • map2sat
  • cityscapes_photo2label
  • cityscapes_label2photo
  • facades_photo2label
  • facades_label2photo
  • iphone2dslr_flower

미리만들어진 모델 테스트

python test.py --dataroot datasets/horse2zebra/testA --name horse2zebra_pretrained --model test