如何在可可数据集中定义内部多边形?

时间:2019-12-30 17:12:52

标签: python-3.x tensorflow conv-neural-network faster-rcnn semantic-segmentation

我想了解如何在可可数据集的分割部分中定义内部或内部多边形

我想教卷积网络识别建筑物多边形中的孔

带孔多边形的例子

enter image description here

1 个答案:

答案 0 :(得分:2)

最好的解决方案是使用RLE(行程编码)蒙版定义带有孔的形状。

在Matterport遮罩R-CNN实现中,所有多边形分割都将转换为RLE,然后再转换为遮罩。在coco.py中签出annToMask()和annToRLE()。多边形的原因是它们在json中存储的效率更高,并且会缩小注释文件的大小。如果无法用实心多边形定义形状,则可能会遇到更大的RLE,除非您想在COCO中创建自己的自定义注释并修改神经网络以自行将这些多边形孔转换为RLE。 / p>

请注意,如果您使用的是仅能找到边界框的神经网络,那么这一切都是不必要的,因为它始终无法返回孔。