如何重新训练一个目标检测模型来检测除了已经训练过的类之外的新类目标?

2024-04-24 23:48:49 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经成功地用我的自定义标签图像重新训练了一个目标检测模型,使用labelImg工具(ssd\u mobilenet\u v1\u coco)检测猫,但它没有检测到它预先训练过的对象(90个类)

我想要一种再培训的方法,这样它可以训练猫和其他90个预先训练的课程。

我尝试在label.pbtxt中添加id:91的新标签,并在生成记录文件时添加与返回值相同的id

   def class_text_to_int(row_label):
    if row_label == 'cat':
        return 91

我跟随this链接进行再培训


Tags: 工具对象模型图像id目标标签label