博客

分割中将mask与原图叠加，生成漂亮的掩膜。

import os
from PIL import Image

    
root_path_background = "/home/cc/codes/python/YOLOP-main/datas/images/train/"
root_path_paste = "/home/cc/codes/python/YOLOP-main/datas/da_seg_annotations/train/"
output_path = "/home/cc/codes/python/YOLOP-main/datas/masks/"
img_list = os.listdir(root_path_background)
label_list = os.listdir(root_path_paste)
img_list = sorted(img_list)
label_list = sorted(label_list)
for num,img_label in enumerate(zip(img_list,label_list)): 
    img = Image.open(os.path.join(root_path_background,img_label[0]))
    label = Image.open(os.path.join(root_path_paste,img_label[1]))
    label = label.convert("RGB")
    fin = Image.blend(img,label,0.3)
    # fin.show()
    fin.save(os.path.join(output_path,img_label[0]))