opencv实现图像去除单一颜色背景

思路

因为背景是固定颜色，很容易筛选出背景，然后将其设为白色完全透明即可。

代码

#coding=utf-8

import cv2 as cv

bg_color = [197, 102, 6]
threshold = 3000

def calc_diff(pixel):
'''
计算pixel与背景的离差平方和，作为当前像素点与背景相似程度的度量
'''
    return (pixel[0]-bg_color[0])**2 + (pixel[1]-bg_color[1])**2 + (pixel[2]-bg_color[2])**2

def remove_bg():
    image_path = './logo.png'
    logo = cv.imread(image_path)
    logo = cv.cvtColor(logo, cv.COLOR_BGR2BGRA) #将图像转成带透明通道的BGRA格式
    h, w = logo.shape[0:2]
    for i in range(h):
        for j in range(w):
            if calc_diff(logo[i][j]) < threshold:
            #若果logo[i][j]为背景，将其颜色设为白色，且完全透明
                logo[i][j][0] = 255
                logo[i][j][1] = 255
                logo[i][j][2] = 255
                logo[i][j][3] = 0
	
    cv.imwrite("./logo_rmbg.png", logo)
                
if __name__ == '__main__':
    remove_bg()

使用方法

修改第5行的bg_color为图片背景的bgr值，以及第6行的threshold（threshold越大，覆盖的像素越多）。

效果：

emmm，事实证明背景附近的颜色不是严格的背景色，后来将字填充后好多了。