From a86fa495e19c79e16a2b0184abd6992abaa0ac48 Mon Sep 17 00:00:00 2001 From: junyanz Date: Thu, 20 Apr 2017 03:30:08 -0700 Subject: support different sizes of dataset A/B and update README --- scripts/test_pix2pix.sh | 2 +- scripts/train_cyclegan.sh | 2 +- scripts/train_pix2pix.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) (limited to 'scripts') diff --git a/scripts/test_pix2pix.sh b/scripts/test_pix2pix.sh index 7b056fe..f46b262 100644 --- a/scripts/test_pix2pix.sh +++ b/scripts/test_pix2pix.sh @@ -1 +1 @@ -python test.py --dataroot=./datasets/facades --name facades_pix2pix --model pix2pix --align_data +python test.py --dataroot=./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --align_data --which_direction BtoA diff --git a/scripts/train_cyclegan.sh b/scripts/train_cyclegan.sh index 03f7fd9..a3dd29b 100644 --- a/scripts/train_cyclegan.sh +++ b/scripts/train_cyclegan.sh @@ -1 +1 @@ -python train.py --dataroot=./datasets/horse2zebra --name horse2zebra_cyclegan --gpu_ids 0 --save_epoch_freq 5 +python train.py --dataroot=./datasets/maps --name maps_cyclegan --save_epoch_freq 5 diff --git a/scripts/train_pix2pix.sh b/scripts/train_pix2pix.sh index 682c6c6..b96517b 100644 --- a/scripts/train_pix2pix.sh +++ b/scripts/train_pix2pix.sh @@ -1 +1 @@ -python train.py --dataroot=./datasets/facades --name facades_pix2pix --which_model_netG resnet_9blocks --loadSize 286 --fineSize 256 --model pix2pix --align_data --which_direction BtoA --save_epoch_freq 25 +python train.py --dataroot=./datasets/facades --name facades_pix2pix --which_model_netG unet_256 --loadSize 286 --fineSize 256 --model pix2pix --align_data --which_direction BtoA --save_epoch_freq 25 -- cgit v1.2.3-70-g09d2