pix2pix訓練

ok，聽別人的建議做的。

1、下載代碼：

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

2、環境配置

 

 3、數據

我的數據

人家數據

數據下載路徑：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、原始模型的訓練（人家是啥樣，我們就啥樣，不做調整的）

a、模型計算量統計

b、代碼我做了修改的地方

c、訓練命令

python train.py --dataroot ./datasets/people --name facades_pix2pix --model pix2pix  --continue_train  --batch_size 50

d、訓練心得

人家說用樣本對的方法，訓練樣本最好處理成樣本對，不然的話，訓練真的是一言難盡。

如何把樣本整理成樣本對。參見我的另一篇博客https://www.cnblogs.com/wjjcjj/p/12574063.html

我就是這么做的。