如何從圖像中屏蔽掉 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 具有正確的值范圍，那么您乘以...可能會起作用（倒置除外）。