OpenCV的Python的 - 设置背景颜色

Question

我想从照片删除灰色背景和一个白色的

取代它到目前为止，我有这样的代码：

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 

的问题是，图像现在有一个黑色的背景，因为我已经掩盖了灰色。我怎样才能用白色的替换空像素？

Answer 1

而不是使用bitwise_not的，我会用

resized.setTo([255, 255, 255], mask) 

这样做，我也想削弱和扩张和面具，在面具摆脱规格的在此之前，是您想要保留的图像的一部分。 http://docs.opencv.org/doc/tutorials/imgproc/erosion_dilatation/erosion_dilatation.html

Answer 2

您可以使用面膜索引数组，并指定只是面具的白色部分为白色：

coloured = resized.copy() 
coloured[mask == 255] = (255, 255, 255) 

Answer 3

我真的建议你坚持用OpenCV的，它是很好的优化。诀窍是反转蒙版并将其应用于某些背景，您将获得蒙版图像和蒙版背景，然后将两者结合。 image1是用原始蒙版蒙版的图像，image2是用倒转蒙版蒙版的背景图像，image3是合成图像。 重要。 image1，image2和image3必须具有相同的大小和类型。蒙版必须是灰度。

import cv2 
import numpy as np 

# opencv loads the image in BGR, convert it to RGB 
img = cv2.cvtColor(cv2.imread('E:\\FOTOS\\opencv\\zAJLd.jpg'), 
        cv2.COLOR_BGR2RGB) 
lower_white = np.array([220, 220, 220], dtype=np.uint8) 
upper_white = np.array([255, 255, 255], dtype=np.uint8) 
mask = cv2.inRange(img, lower_white, upper_white) # could also use threshold 
mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))) # "erase" the small white points in the resulting mask 
mask = cv2.bitwise_not(mask) # invert mask 

# load background (could be an image too) 
bk = np.full(img.shape, 255, dtype=np.uint8) # white bk 

# get masked foreground 
fg_masked = cv2.bitwise_and(img, img, mask=mask) 

# get masked background, mask must be inverted 
mask = cv2.bitwise_not(mask) 
bk_masked = cv2.bitwise_and(bk, bk, mask=mask) 

# combine masked foreground and masked background 
final = cv2.bitwise_or(fg_masked, bk_masked) 
mask = cv2.bitwise_not(mask) # revert mask to original 

Answer 4

首先，你需要得到的背景。必须用掩模图像从原始图像中减去此值。然后将黑色背景更改为白色（或任何颜色）。然后回到添加面具的图像。 看这里OpenCV grabcut() background color and Contour in Python