pix2pix使用自己数据集

pix2pix使用自己数据集时，需要通过combine a and b来进行，从开始到结束出现了几个错误，
1.是多进程，freeze相关，pool限制一下processes=60就可以（processes<63）

2.不知道为什么进行combine时又重新生成了一个同名的文件夹，也没有细看，开始debug了半天没有发现错误，后来才发现是，又重新建立了一个完全同名的文件夹里面只有fold_AB。在此期间以为没有拼接成果，用了一个简单版的代替，但没法多进程

combine_ziji.py
 

import os
import cv2
import numpy as np

def image_write(path_A, path_B, path_AB):
    im_A = cv2.imread(path_A, 1)
    im_B = cv2.imread(path_B, 1)
    im_AB = np.concatenate([im_A, im_B], 1)
    cv2.imwrite(path_AB, im_AB)

def main():
    folder_A = 'path_to_A_folder'  # 替换为 A 文件夹的路径
    folder_B = 'path_to_B_folder'  # 替换为 B 文件夹的路径
    folder_AB = 'path_to_AB_folder' # 替换为输出文件夹的路径

    if not os.path.exists(folder_AB):
        os.makedirs(folder_AB)

    for filename in os.listdir(folder_A):
        if filename.endswith('.tif'):
            path_A = os.path.join(folder_A, filename)
            path_B = os.path.join(folder_B, filename)
            path_AB = os.path.join(folder_AB, filename)
            if os.path.exists(path_B):
                image_write(path_A, path_B, path_AB)

if __name__ == '__main__':
    main()