我使用 torch 和一些语义分割算法来生成分割图像的二进制掩码。然后我想根据该蒙版裁剪图像。为了清楚起见，我需要在每个像素的基础上裁剪它。这似乎是一个简单的问题，但我能想到的唯一解决方案是反转draw mask类似于 Coco API 中的函数，或者迭代数组中的每个像素，并在不需要时一起将像素设置为黑色。我觉得有更好的方法可以做到这一点。 Lua、Python、Go 或 C++ 的库适合我。有任何想法吗？
我已经在 Python 中实现了这一点，假设您有输入图像和遮罩作为 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 的部分。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
