Pix2Pix
这里是基于Pytorch的pix2pix实现的github链接GitHub - junyanz/pytorch-CycleGAN-and-pix2pix: Image-to-Image Translation in PyTorch这次课程是通过华为昇思Mindspore框架来进行pix2pix模型的训练和推理,这里有官方提供的数据集,这次的数据集经过处理,可以直接使用indspore.dataset的方法读取
生成器(使用U-Net结构)和判别器(使用PatchGAN结构),训练的时候 get_lr
函数定义了一个动态调整的学习率调度,开始时使用固定的学习率 lr
,在经过 n_epochs_decay
个周期后,学习率会线性衰减到0,然后加载的数据集是官方提供好的,运行jupyterLab的cell就可以训练了,训练完再进行推理
结果