python怎么更改背景颜色,OpenCV Python-设置背景颜色

I am trying to remove a greyish background from a photo and replace it with a white one

so far I have this code:

image = cv2.imread(args["image"])

r = 150.0 / image.shape[1]

dim = (150, int(image.shape[0] * r))

resized = cv2.resize(image, dim, interpolation=cv2.INTER_AREA)

lower_white = np.array([220, 220, 220], dtype=np.uint8)

upper_white = np.array([255, 255, 255], dtype=np.uint8)

mask = cv2.inRange(resized, lower_white, upper_white) # could also use threshold

res = cv2.bitwise_not(resized, resized, mask)

cv2.imshow('res', res) # gives black background

The problem is that the image now has a black background as I have masked out the grey. How can I replace the empty pixels with white ones?

解决方案

You can use the mask to index the array, and assign just the white parts of the mask to white:

coloured = resized.copy()

coloured[mask == 255] = (255, 255, 255)