pix2pix训练

时间：2020-03-25 17:35:41 收藏：0 阅读：316

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