本项目是基于官方 Pix2PixHD 实现的有监督图像翻译
- 将两个数据域的图像以如下结构部署
---data_root
------A
------B
- 训练
# example
python train.py --data_root xxx --sample ./sample --img_size 256
# 更多参数指定详情请看./options/trainOption.py
- 测试
python train.py --test_dir xxx --save_dir xxx --img_size 256
- model: 存放神经网络机构
- option: 用于定义超参数
- output: 存放训练过程的中间结果
- pretrained_checkpoint: 存放预训练模型
This code borrows heavily from Pix2PixHD.