pix2pix训练自己的训练集

首先是官方代码：GitHub - junyanz/pytorch-CycleGAN-and-pix2pix: Image-to-Image Translation in PyTorch

配置相关环境可以参考其他文章，本篇在环境配置好的情况下进行

数据集模式：pix2pix需要成对的数据集，官方代码将两个不同模态的图像进行拼接的代码本人无法实现，这里给出自己的代码；

import skimage.io as io
import numpy as np
import os

dir_path1 = "datasets/FS2K/FSGAN256/A/photo3"

dir_path2 = "datasets/FS2K/FSGAN256/B/sketch3"

files = os.listdir(dir_path1)

for file in files:
    im1 = io.imread(dir_path1 + '/' + file)  # np.ndarray, [h, w, c], 值域(0, 255), RGB
    # plt.imshow(im1)   #查看图片
    # plt.show()

    im2 = io.imread(dir_path2 + '/' + file)  # np.ndarray, [h, w, c], 值域(0, 255), RGB
    # plt.imshow(im2)
    # plt.show()

    # print(im1.shape)   #查