机器视觉-使用opencv将人像抠出来,加上纯色背景图

背景

需求:使用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
上一篇:Linux系统进程管理和性能优化


下一篇:CF715B Complete The Graph 题解