用Python对两个数据集中的图像进行水平拼接

时间:2022-07-28
本文章向大家介绍用Python对两个数据集中的图像进行水平拼接,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

交流、咨询,有疑问欢迎添加QQ 2125364717,一起交流、一起发现问题、一起进步啊,哈哈哈哈哈

各位看官老爷,如果觉得对您有用麻烦赏个子,创作不易,0.1元就行了。下面是微信乞讨码:

添加描述

添加描述

A图:

B图:

拼接后:

import os
import numpy as np
import PIL
from PIL import Image

dirname_read_A = "D:paper\3low_light_imagedataset\train_translste_model_dataimage_translate\trainA_concat\"
names_A=os.listdir(dirname_read_A)
names_A.sort(key= lambda x:int(x[:-6]))

dirname_read_B = "D:paper\3low_light_imagedataset\train_translste_model_dataimage_translate\trainB_concat\"
names_B=os.listdir(dirname_read_B)
names_B.sort(key= lambda x:int(x[:-6]))

k = 1

for i in range(2940):
   imgA = dirname_read_A + names_A[i]
   imgB = dirname_read_B + names_B[i]

   imgA = Image.open(imgA)
   imgB = Image.open(imgB)

   img_array_A = np.array(imgA)
   img_array_B = np.array(imgB)

   img_array_AB = np.hstack((img_array_A, img_array_B))
   # print(img_array_A.shape)
   # print(img_array_B.shape)
   print(img_array_AB.shape)
   imgAB = Image.fromarray(img_array_AB.astype('uint8')).convert('RGB')
   imgAB.save(str(k) + '.jpg')

   k = k + 1