首先是官方代码: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) #查