除了已训练的类别外,如何重新训练对象检测模型以检测新的对象类别?

时间:2019-12-27 08:01:08

标签: python tensorflow object-detection pre-trained-model

我已使用labelImg工具(ssd_mobilenet_v1_coco)成功地使用自定义标签图像重新训练了对象检测模型,以检测猫,但无法检测到预先训练过的对象(90类)。

我想要一种重新训练的方式,以便它训练猫和其他90个预先训练的班级。

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

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

我已经按照this链接进行了重新培训。

0 个答案:

没有答案