需要将掩膜叠加在原图上,并调整掩膜至半透明状态。试了很久,终于找到了最简单最有效的办法。简单做个记录,如果有什么问题请各位大牛指正。
主要使用了cv2.addWeighted函数。
融合的两个图像大小需要相同。参考cv2.addWeighted
想强调的是Python在处理图像时,有各种模块可以调用。但是各模块在读取和处理图像时各有特色。所以经常出现混合用的情况。比如cv2读取图像后为矩阵,可以直接修改图像。这里特意用了两个函数,一个函数fcv主要使用cv2模块,一个函数mask使用PIL的Image。函数mask可对掩膜的颜色进行修改,掩膜位置变色,黑色背景部分变透明。具体见不同图像模块
还有一个问题就是png和jpg图像在读取时也不同,png为RGBA四通道,转化为矩阵后,图像宽和高也发生了翻转。
import cv2,os
from PIL import Image
imgpath='C:\\brc_b.png'
imgpath_mask='C:\\brc_b_m.png'
def fcv(imgpath):#将原图和掩膜融合
mask_path = mask(imgpath_mask)
mask = cv2.imread(mask_path)
img_o = cv2.imread(imgpath)
alpha=0.5
beta=1
output=img_o.copy()
print(type(output))
cv2.addWeighted(mask, alpha, output, beta,0)
cv2.imshow("output",output)
cv2.waitKey(0)
def mask(imgpath_mask):#保存掩膜,返回掩膜地址
img_m = Image.open(imgpath_mask)#图片为png格式,所以不需要转换为RGBA格式
#img_m = img_m.convert('RGBA')#若非png格式,需要转化格式
h_m,w_m=img_m.size#一般情况下为(w,h),png四通道格式下,为(h,w)
red_pixel=(0,255,0,0) #green_pixel=(255,0,0,0),yellow_pixel=(255,255,0,0),black_pixel=(0,0,0,255),white_pixel=(255,255,255,255) 常见颜色
for i in range(int(h_m)):
for j in range(int(w_m)):
if img_m.getpixel((i,j))==black_pixel:#掩膜中黑色的像素变透明
img_m.putpixel((i,j),(0,0,0,0))
elif img_m.getpixel((i,j))==white_pixel:#掩膜中白色的像素变成黄色
img_m.putpixel((i,j),(255,255,0,0))
print('MASKdone')
image_save_path = 'C:\\brc-b-m.png' ##将整数和字符串连接在一起
img_m.save(image_save_path)
return image_save_path
fcv(imgpath)
print('done')
原图
背景不透明时,可设置为(0,0,0,255)
背景不透明时,可设置为(0,0,0,0)
最后融合透明背景的掩膜和原图后的结果