对象检测MASK RCNN仅适用于2类

Question

我使用（https://github.com/matterport/Mask_RCNN）MASKRCNN进行对象检测，并且一切正常。有没有办法只检测某些对象？我有80多个课程，我只需要2个课程（例如汽车和人）。我想剩下的类没有被检测到。如何将其删除？

Answer 1

我猜您正在使用默认的预训练可可模型进行检测，该模型具有80个类别。您可以训练自己的模型。 首先，您必须使用VIA（VGG图像注释器）来标记您要预测的类。 完成后，您必须对模型代码进行一些更改。例如，如果您使用文件“ balloon.py” ，则必须添加类，并在 load_mask（） 函数中进行一些更改以及代码的其他部分。之后，您可以开始训练模型，然后用于检测和细分。

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

检查一个示例here