合并来自两个differents文件夹的两个图像(python)

时间:2020-04-20 17:37:27

标签: python numpy opencv

我正在尝试将来自两个不同文件夹的所有图像一一合并,例如,我拥有这两个文件夹

-folder1

-+ img1.jpg

-+ img2.jpg

-+ img3.jpg

...

-folder2

-+ img_1.jpg

-+ img_2.jpg

-+ img_3.jpg

...

我想做的就是将img1.jpg和img_1.jpg,img2.jpg和img_2.jpg结合起来...

我正在尝试使用此代码来使其正常工作,但问题是仅从文件夹中获取相同的图像,并将其与其他文件夹中的所有图像组合在一起

def load_images_from_folder(folder,folder2):
images = []
i=0
for filename in os.listdir(folder):
    for filename2 in os.listdir(folder2):
        img1 = cv2.imread(os.path.join(folder,filename))
        img2 = cv2.imread(os.path.join(folder2,filename2))
        img2 = np.fliplr(img2)
        vis = np.concatenate((img1, img2), axis=1)

有什么想法可以使此代码正常工作吗?

1 个答案:

答案 0 :(得分:1)

您可以使用zip

def load_images_from_folder(folder,folder2):
    images = []
    i=0
    for filename, filename2 in zip(os.listdir(folder), os.listdir(folder2)):
        img1 = cv2.imread(os.path.join(folder,filename))
        img2 = cv2.imread(os.path.join(folder2,filename2))
        img2 = np.fliplr(img2)
        vis = np.concatenate((img1, img2), axis=1)
相关问题