合并来自不同文件夹的图像并将其存储到其他文件夹

时间:2020-06-30 00:21:07

标签: python opencv merge image-preprocessing

我有三个不同的文件夹m1,m2和m3。 “ m1”文件夹包含image(i)_m1.png格式的图像(其中i = 1到N),“ m2”文件夹包含image(i)_m2.png格式的图像,“ m3”文件夹包含图像格式为image(i)_m3.png。我想像这样使用cv2.merge合并这些图像:(cv2.merge((image1_m1,image1_m2,image1_m3))并且它持续N次并存储在与包含'N'个格式图像的合并图像不同的文件夹中(i)_merged.png。

import pandas as pd
import cv2
import numpy as np
import glob
import os

filenames1 = glob.glob("data_folder/m1/*.png")
filenames1.sort()
filenames2 = glob.glob("data_folder/m2/*.png")
filenames2.sort()
filenames3 = glob.glob("data_folder/m3/*.png")
filenames3.sort()


for f1 in filenames1:
    for f2 in filenames2:
        for f3 in filenames3:
            img1 = cv2.imread(f1)
            img2 = cv2.imread(f2)
            img3 = cv2.imread(f3)
            img_m1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
            img_m2 = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
            img_m3 = cv2.cvtColor(img3, cv2.COLOR_BGR2GRAY)
            img_rgb = cv2.merge((img_m1, img_m2, img_m3))
            cv2.imwrite("data_folder/merge/img_name.png", img_rgb)

1 个答案:

答案 0 :(得分:1)

您的问题不完整。我假设您对for循环有问题。 您可以将嵌套的for循环替换为此:

for f1,f2,f3 in zip(filenames1,filenames2,filenames3):
相关问题