我使用(https://github.com/matterport/Mask_RCNN)MASKRCNN进行对象检测,一切正常。有没有办法只检测某些对象?我有80多个课程,我只需要2个课程(例如汽车和人)。我想剩下的类没有被检测到。如何将其删除?
我猜您正在使用默认的预训练可可模型进行检测,其中包含80个类别。您可以训练自己的模型。首先,您必须使用VIA(VGG图像注释器)来标记您要预测的类。完成后,您必须对模型代码进行一些更改。例如,如果您使用文件“ balloon.py”,则必须添加类,在load_mask()函数中进行一些更改,而代码的其他部分则很少。之后,您可以开始训练模型,然后将其用于检测和细分。
是的,对于VIA,请尝试使用1.0.0版本,因为更新后的版本中.json文件的格式略有变化,这通常使它们与自定义数据集的训练不兼容。
在这里检查一个例子
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句