image - 如何基于二进制掩码裁剪图像

Question

我正在使用 torch 和一些语义分割算法来生成分割图像的二进制掩码。然后我想根据该蒙版裁剪图像。明确地说，我需要在每个像素的基础上裁剪它。这似乎是一个简单的问题，但我能想到的唯一解决方案是像 Coco API 中那样反转 draw mask 函数，或者遍历数组中的每个像素并将像素设置为黑色如果不需要。我觉得有更好的方法可以做到这一点。 Lua、Python、Go 或 C++ 中的库适合我。有什么想法吗？

Answer 1

我已经在 Python 中实现了这一点，假设您的输入图像和 mask 可用作 Mat 对象。鉴于 src1 是您的图像，而 src1_mask 是您的二进制掩码:

src1_mask=cv2.cvtColor(src1_mask,cv2.COLOR_GRAY2BGR)#change mask to a 3 channel image 
mask_out=cv2.subtract(src1_mask,src1)
mask_out=cv2.subtract(src1_mask,mask_out)

现在 mask_out 包含位于您定义的二进制掩码内的图像 src1 部分。