背景

需求：使用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

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