我正在使用torch和一些语义分割算法来生成分割图像的二进制掩膜。然后,我想根据该掩膜裁剪图像。需要明确的是,我需要基于每个像素进行裁剪。这似乎是一个简单的问题,但我能想到的唯一解决方案要么是反转类似于Coco API中的
draw mask
函数,要么是迭代数组中的每个像素并在不需要时将像素设为黑色。我觉得有更好的方法来解决这个问题。对我来说,Lua、Python、Go或C++库都可以。有什么想法吗?