python - 如何使用蒙版去除python中的背景

Question

我想用 mask 图像去除背景。现在，我已经得到了 mask 图像。我试图让原始图像的背景值变为0，其中 mask 值为0。但结果很糟糕。我怎么解决这个问题。谢谢

from skimage import io
import numpy as np
img = io.imread("GT06.jpg")
mask = io.imread("GT03.png")
mask2 = np.where((mask==0),0,1).astype('uint8')
img = img*mask2[:,:,np.newaxis]
io.imshow(img)
io.show()

GT06.jpg

GT03.png

这导致:

我想得到这样的前景:

Answer 1

问题是你的掩码不是纯黑和白的，即所有 0 或 255 将你的掩码两代更改为:

mask2 = np.where((mask<200),0,1).astype('uint8')

结果:

您可以使用掩码或阈值数字 - 我使用了 200。