（学习记录）如何利用语义分割的mask和img进行融合显示

1. 将mask中的白色变为特定颜色

import cv2
import operator

import numpy as np


def img_input(img_path):
    img = cv2.imread(img_path)  # 使用 cv2 来打开图片
    return img


def trans(img):  # 转换颜色
    img2 = img
    Size = img.shape
    width = Size[0]
    length = Size[1]
    num = Size[2]
    list1_red= [255, 255, 255]
    list2_green = [0, 0, 255]
    for i in range(0, width):
        for j in range(0, length):
            if operator.eq(img[i][j].tolist(), list1_red) == True:
                img2[i][j] = np.array([190, 162, 255])#粉色

    return img2


if __name__ == '__main__':
    img_path = "331.png"
    img = img_input(img_path)
    img2 = trans(img)
    cv2.imwrite(filename="./1_331.png", img=img2)

2. 将黑色背景转化为透明背景

import cv2
import numpy as np

path = "1_331.png"
src = cv2.imread(path)

# Point 1: 生成与黑色部分对应的mask图像
mask = np.all(src[:, :, :] == [0, 0, 0], axis=-1)

# Point 2: 将图片从三通道转为四通道
dst = cv2.cvtColor(src, cv2.COLOR_BGR2BGRA)

# Point3:  以mask图像为基础，使黑色部分透明化
dst[mask, 3] = 0

# 保存图片
cv2.imwrite("2_331.png", dst)

3. mask和img相加得到融合

import PIL.Image as Image

layer1 = Image.open("IMG_0331.JPG").convert('RGBA')   # 底图背景
layer2 = Image.open("2_331.png").convert('RGBA')    # mask


final = Image.new("RGBA", layer1.size)             # 合成的image
final = Image.alpha_composite(final, layer1)
final = Image.alpha_composite(final, layer2)

final=final.convert('RGB')
final.save('3_331.jpg')

第3步还有一种融合显示，不过背景色会变暗

import cv2
 
img1=cv2.imread('ml.png') 
img2=cv2.imread('opencv_logo.jpg')
dst=cv2.addWeighted(img1,0.7,img2,0.3,0) 
cv2.imwrite("dst.jpg",dst)