文件夹下同名图片mask、mask_1叠加

 

#In[]合并后缀mask和mask1读取文件夹中的图片
from PIL import Image
import os

# 遍历文件夹获取所有"mask.png"和"mask_1.png"
root_dir = '1'  # 假设当前目录下有这些文件
files = os.listdir(root_dir)

pairs = [(file, file.replace('.png', '_1.png')) for file in files if file.endswith('mask.png')]

for pair in pairs:
    mask0 = cv2.imread(os.path.join(root_dir, pair[0]))
    mask1 = cv2.imread(os.path.join(root_dir, pair[1]))
    if mask1 is None:
         continue
    # 叠加图像
    image= mask0 + mask1
    # 定义输出文件名，比如组合后的图片命名为"combined_mask.png"
    output_file = pair[0].replace('.png', '_combined.png')

    # 保存合并后的图片
    cv2.imwrite(output_file,image)