不同文件夹下的图像拼接在一起

最近在研究论文“Toward Multimodal Image-to-Image Translation”的时候：Toward Multimodal Image-to-Image Translation

发现他们的输入是两张256*256的照片横着拼在一下形成512*256的照片作为输入，所以我这里实现了将不同文件夹下的图像拼接在一起

备注：其中A与B文件夹下的照片文件名是一致的，也就是一一对应的

---

import cv2
import glob
import os
import numpy as np

# 文件夹A下的所有照片
input_img_domainA = glob.glob("./molecular_bigdataset/train/A/*.png")

save_path = "new_generate_image/"


def mkdir(path):
    folder = os.path.exists(path)
    if not folder:  # 判断是否存在文件夹如果不存在则创建为文件夹
        os.makedirs(path)  # makedirs 创建文件时如果路径不存在会创建这个路径
        print("--- create new folder...  ---")
    else:
        print("---  There is this folder!  ---")

mkdir(save_path)  # 调用函数


i = 0
for file in input_img_domainA:
    # get the file_name of image
    file_name = file.split('\\')[-1]

    img1 = cv2.imread(file)
    # 文件夹B下的所有照片
    img2 = cv2.imread('./molecular_bigdataset/train/B/'+ file_name)

    image = np.concatenate([img1, img2], axis=1)

    cv2.imwrite(save_path + file_name, image)

    i = i + 1
    print("The", i, "picture is currently being processed")