fix cyclegan infer (#2588)

379294ef · lvmengsi · GitHub · 336c8412 · 379294ef
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleCV/gan/scripts/infer_cyclegan.sh PaddleCV/gan/scripts/infer_cyclegan.sh +1 -1

未找到文件。
--- a/PaddleCV/gan/scripts/infer_cyclegan.sh
+++ b/PaddleCV/gan/scripts/infer_cyclegan.sh
-python infer.py --init_model output/checkpoints/199/ --dataset_dir "data/cityscapes/testA/*" --image_size 256 --input_style A --model_net CycleGAN --net_G resnet_6block  --g_base_dims 32
+python infer.py --init_model output/checkpoints/199/ --dataset_dir "data/cityscapes/testB/*" --image_size 256 --input_style B --model_net CycleGAN --net_G resnet_9block  --g_base_dims 32