我正在尝试在张量流对象检测模型中向ms_coco_model(具有90个类)添加新类,并运行相同的训练集 . 我正在使用预先训练的模型检查点:“ssd_mobilenet_v1_coco_checkpoint”来加速训练步骤 .
我关注了博客https://towardsdatascience.com/how-to-train-your-own-object-detector-with-tensorflows-object-detector-api-bec72ecfe1d9,这很有效 . 因为它仅在.pbtxt文件中具有1个类的浣熊数据集上运行训练步骤 . 输出与预期一致,能够检测给定输入文件中的浣熊"only" .
我想要实现的是,能够在给定的输入文件中检测浣熊以及其他对象 . 我按照以下方法:
-
将浣熊注释.xml转换为.json格式(与ms_coco数据集相同)
-
为ms_coco图像和注释文件添加了浣熊图像和注释
-
使用create_tfrecord脚本为组合的ms_coco浣熊数据生成tfrecord,用于训练步骤
-
更新了.pbtxt文件并添加了id为91的raccoon类
-
使用预先训练的模型ssd_mobilenet_v1_coco_checkpoint开始训练步骤
但即使经过约2k步的训练,我也无法达到预期的输出 .
这种方法是否正确,可以将新类数据添加到现有数据模型中?我在这里错过了吗?