opencv通过mask取目标纯色背景图

import cv2
import numpy as np

'''
    通过掩码图得到纯色背景的目标
'''


rgb = cv2.imread(r'rgb.jpg')
mask = cv2.imread(r'mask.png')

# 第一步：将rgb图的背景区域变为0
black_bg = np.uint8(rgb*(mask/255.))

# 第二步：将掩码原本0的位置改为255，原本255的位置改为0
reversed_msk = 255-mask

# 第三步：将黑色背景位置(像素值为0的位置) 加上255
white_bg = (black_bg + reversed_msk).astype(np.uint8)


# cv2.imshow('blcak_bg', black_bg)
# cv2.imshow('white_bg',white_bg)
# cv2.imwrite('black.jpg',black_bg)
# cv2.imwrite('white.jpg',white_bg)
# cv2.waitKey(0)
new = np.concatenate((rgb,mask,black_bg,white_bg),axis=1)
cv2.imwrite('new.jpg',new)

得到结果：