ok，听别人的建议做的。
1、下载：
https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
2、准备数据
3、环境配置
4、数据
数据下载路径：http://efrosgans.eecs.berkeley.edu/pix2pix/datasets/
人家数据长上面这个样子，好吧，我们的数据也整成这样吧，代码如下：
import os import cv2 import numpy as np Adir = ‘./trainA/‘ Bdir = ‘./trainB/‘ ABdir = ‘./datasets/eye/train/‘ num=0 for fi in os.listdir(Adir): im_A = cv2.imread(Adir+fi) # python2: cv2.CV_LOAD_IMAGE_COLOR; python3: cv2.IMREAD_COLOR im_B = cv2.imread(Bdir+‘0‘+fi[1:]) # python2: cv2.CV_LOAD_IMAGE_COLOR; python3: cv2.IMREAD_COLOR im_AB = np.concatenate([im_A, im_B], 1) cv2.imwrite(ABdir+str(num)+‘.jpg‘, im_AB) num = num + 1 print (num)
5、训练
原文：https://www.cnblogs.com/wjjcjj/p/12567445.html