Pix2Pix 代码理解（PyTorch）updating

论文：https://arxiv.org/abs/1611.07004
代码：https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

训练

train.py入口

import的函数
from options.train_options import TrainOptions
from data import create_dataset
from models import create_model
from util.visualizer import Visualizer

特点
支持不同的模型：pix2pix, cyclegan, colorization
支持不同的数据集模式：aligned, unaligned, single, colorization
训练会保存图片、损失函数可视化、保存模型
可以暂停、继续训练

流程

1. 得到训练的选项opt，使用options.train_options.TrainOptions.parse()函数
2. 根据opt得到数据集dataset，使用data.init.create_dataset(opt)
3. 根据opt得到模型model，使用models.