不同文件夹图片拼接

#导入相应的包
import numpy as np
import os
import matplotlib.image as mp
from skimage import img_as_ubyte
from PIL import Image

#本地图片存放的地址
jpg_path1 = "E:\\BaiduNetdiskDownload\\dataset"
jpg_path2 = "E:\\BaiduNetdiskDownload\\dataset\\rn\\"

#新建文件夹来处理拼接后的新图片
#os.makedirs(" E:\\BaiduNetdiskDownload\\合并后\\")

#拼接处理后的图片保存地址
savepath = r"E:\BaiduNetdiskDownload\dataset\pinjie"

filelist1 = os.listdir(jpg_path1) #打开对应的文件夹
filelist2 = os.listdir(jpg_path2)

total_num = len(filelist1)#得到文件夹中图像的总个数

for i in range(total_num):
    jpg1 = jpg_path1 +'norain'+'-'+ str(i + 1) + '.png' #拼接图像的读取地址
    jpg2 = jpg_path2 +'norain'+ '-'+ str(i + 1) + '.png'

    #对图像数据类型转换
    image1 = mp.imread(jpg1)
    image2 = mp.imread(jpg2)
    image1 = img_as_ubyte(image1)
    image2 = img_as_ubyte(image2)
    result_img1 = Image.fromarray(image1)
    result_img2 = Image.fromarray(image2)

    img1 = np.array(result_img1)  # 转换为多维数组对象
    img2 = np.array(result_img2)
    im3 = np.concatenate((img1, img2), axis=1)  # 对图片进行横向拼接
    # 生成图片
    result_picture = Image.fromarray(im3)
    savepicture_name = savepath + str(i+1) + '.png'  #字符串拼接保存图像的地址
    result_picture.save(savepicture_name)  #保存图像