通过 OpenCV 应用分割掩码

Question

我目前有一个 np.ndarray 格式的位掩码，其中掩码值为 1 的像素和没有掩码的像素值为 0。

我想将其“应用”到另一个 np.ndarray 图像（3 通道：RGB），其中蒙版所在的区域会稍微更加突出显示为突出显示的颜色。例如，如果我想要用绿色表示人体面具的区域，我会想要如下所示的内容。我想知道如何使用 opencv 和 numpy.

来做到这一点

Answer 1

让我们试试cv2.addWeighted:

# sample data
img = np.full((10,10,3), 128, np.uint8)

# sample mask
mask = np.zeros((10,10), np.uint8)
mask[3:6, 3:6] = 1

# color to fill
color = np.array([0,255,0], dtype='uint8')

# equal color where mask, else image
# this would paint your object silhouette entirely with `color`
masked_img = np.where(mask[...,None], color, img)

# use `addWeighted` to blend the two images
# the object will be tinted toward `color`
out = cv2.addWeighted(img, 0.8, masked_img, 0.2,0)

输出：

通过 OpenCV 应用分割掩码

Apply a segmentation mask through OpenCV

python

opencv

numpy

image-segmentation