如何从图像中屏蔽掉 object？

Question

我有一个带有二进制掩码的 256X256 的 RGB 图像。 我想借助掩码从主图像中删除 object 并保留 rest 背景。

我已经尝试过使用下面的代码，但它没有用。

谁能让我知道我在哪里犯了错误？

import cv2
import numpy as np

# read image
img = cv2.imread('image.jpg')

mask2 = cv2.imread('mask.jpg') / 1
# mask by multiplication, clip to range 0 to 255 and make integer
result2 = (img * mask2).clip(0, 255).astype(np.uint8)



# save results
cv2.imwrite('result.png', result1)

图片，面具：

Output 我想要，我得到的结果：

Answer 1

您需要使用cv2.bitwise_and来解决您的问题。 它以按位方式在您的img和mask之间执行“与”操作。

• 对于mask中的白色区域，保留对应的img区域
• 对于mask中的白色区域，对应的img区域被遮挡

代码：

img = cv2.imread(r'C:\Users\524316\Desktop\Stack\eye\eye.jpg')
mask = cv2.imread(r'C:\Users\524316\Desktop\Stack\eye\mask.jpg', 0)

# invert the mask
th = cv2.threshold(mask,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)[1]

# preserve the background
res = cv2.bitwise_and(img, img, mask = th)

结果：

Answer 2

您不需要 Otsu 阈值，只需普通阈值，具有合理的阈值。 您的“掩码”图像是 JPEG。 有损压缩会导致它不完全是黑色或白色。

有了遮罩后，您只需要了解遮罩操作即可。 这是我在这里使用的numpy的一个功能：

im = cv.imread("VgAjK.jpg")
maskim = cv.imread("ZKKkp.jpg", cv.IMREAD_GRAYSCALE)

im[maskim >= 128] = 255 # "remove", paint over with white

如果您的 arrays 具有正确的值范围，那么您乘以...可能会起作用（倒置除外）。