文件夹下同名图片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)