- 将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)
- 将黑色背景转化为透明背景
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)
- 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)