背景
需求:使用opencv将人像抠出来,加上纯色背景图,网上找遍资源没有找到合适的办法,故使用最笨的方法实现。
def change_bg(mask, fg, bg):
"""
:param mask: mask np.array图对象 mask和fg等高等宽等通道,八位二值灰度图
:param fg: 人像白色背景图 np.array图对象 mask和fg等高等宽等通道
:param bg: 纯色背景图 np.array图对象 大小无所谓,RGB模式的图
:return: 合成图 np.array图对象
"""
# 遍历替换
w, h = mask.shape
bg = np.resize(bg, fg.shape)
for j in range(h):
for i in range(w):
if mask.item(i, j) == 255:
# 当读取到mask白色像素时,人像白色背景图上的像素赋值给纯色背景图上同一位置
bg.itemset((i, j, 0), fg.item(i, j, 0)) # 此处替换颜色,为BGR通道
bg.itemset((i, j, 1), fg.item(i, j, 1)) # 此处替换颜色,为BGR通道
bg.itemset((i, j, 2), fg.item(i, j, 2)) # 此处替换颜色,为BGR通道
# bg[i,j]=fg[i,j] # 或者使用这个也行,根据自己业务场景来做
return bg