在新对象类的训练Yolov3模型中需要帮助

时间:2019-06-22 08:12:24

标签: machine-learning computer-vision object-detection yolo

如何使用自定义数据在新类上训练对象检测模型?

我想在两个类Persondustbin上训练yolov3模型。由于有些砝码已经在包括人在内的80个不同类别中进行过训练,因此我想使用该经过预先训练的模型对我的新班级垃圾箱进行训练。我已经废弃了约1000张垃圾桶的图像并贴了标签。我很困惑。

  1. 由于该模型已经过人身培训,因此我需要给它提供类别人的标签数据吗?或者我可以只使用标记为Dustbin的数据,它将(希望)开始检测垃圾箱和人员(如基本模型)

我已经对看起来像这样的coco.names文件进行了更改

person
dustbin

,并且还对cfg文件进行了必要的更改,即将过滤条件更改为21。

很抱歉,对于天真的问题,我才刚刚开始学习有关对象检测和yolo的知识,并尽力按照stackoverflow的要求来格式化问题。谢谢

0 个答案:

没有答案