对象检测掩码 RCNN 仅适用于 2 个类



我使用(https://github.com/matterport/Mask_RCNN(MASKRCNN进行对象检测,一切正常。有没有办法只检测某些物体?有80多个类,我只需要2个(如汽车和人(。我希望不检测到其余的类。如何删除它们?

我猜你正在使用默认的预训练coco模型进行检测,该模型带有80个类。您可以训练自己的模型。 首先,您必须使用VIA(VGG图像注释器(来标记要预测的类别。 完成此操作后,您必须对模型的代码进行一些更改。例如,如果您使用的是文件"balloon.py">,则必须添加类,对load_mask((函数和代码的其他部分进行一些更改。之后,您可以开始训练模型,然后用于检测和分割。

是的,对于 VIA,请尝试使用1.0.0版本,因为在更新版本中.json文件的格式略有变化,这通常会使它们与自定义数据集的训练不兼容。

在此处查看一个示例

相关内容

  • 没有找到相关文章

最新更新