我已使用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链接进行了重新培训。