python实现深度学习中原始图像与预测结果叠置输出

python实现深度学习中原始图像与预测结果叠置输出

讲解
略
代码

from PIL import Image


def add_alpha_channel(img):
    img = Image.open(img)
    img = img.convert('RGBA')
    # 更改图像透明度
    # factor = 0.7
    # img_blender = Image.new('RGBA', img.size, (0, 0, 0, 0))
    # img = Image.blend(img_blender, img, factor)
    return img


def image_together(image, layer, save_path, save_name):
    layer = layer
    base = image
    bands = list(layer.split())
    heigh, width = layer.size
    for i in range(heigh):
        for j in range(width):
            r, g, b, a = layer.getpixel((i, j))
            if r == 0:
                layer.putpixel((i, j), (0, 0, 0, 0))
            else:
                layer.putpixel((i, j), (256, 0, 0, 200))
    base.paste(layer, (0, 0), layer)  # 贴图操作
    base.save(save_path + "/" + save_name + ".png")  # 图片保存


if __name__ == "__main__":
    image = add_alpha_channel(r"")
    layer = add_alpha_channel(r"")
    image_together(image, layer, r"", "")

运行结果
运行前


运行后