Created by: shaoshenchen

在第一个项目中预测，这里训练时使用的是batch_norm

%cd /home/aistudio/gan
!python infer.py \
  --model_net="CycleGAN" \
  --init_model="output/checkpoints/99" \
  --input_style="B" \
  --n_samples=1 \
  --dataset_dir="my_dataset" \
  --test_list="my_dataset/labels.txt" \
  --image_size=256 \
  --crop_size=256 \
  --output="my_dataset/output" \
  --use_gpu=False

最后几行输出结果：

output/checkpoints/99/net_G

load params done

my_dataset my_dataset/labels.txt

read: ['hiking.jpg']

---

第二个项目预测，训练使用的是instance_norm

%cd /home/aistudio/gan
!python infer.py \
    --model_net="CycleGAN" \
    --init_model="output/checkpoints/30" \
    --input_style="B" \
    --output="my_dataset/output" \
    --dataset_dir="my_dataset" \
    --test_list="my_dataset/labels.txt" \
    --norm_type="instance_norm"

最后几行输出结果：

output/checkpoints/30/net_G

load params done

my_dataset my_dataset/labels.txt

---

我把两个项目的地址公开在了AI Studio

刚才用cpu测试了一下第一个项目， labels.txt 中两张图片只能得到一张预测结果；

用cpu测试了一下第二个项目，没有出结果。