重新训练Tensorflow对象检测API

Question

我已经阅读了有关如何使用TensorFlow对象检测API训练新类的教程。但是我想做的是在预训练模型的已经训练好的课程中添加一个新的课程。

例如：MS-COCO预训练模型有90个班级。我想再添加一个类并检测91个类的对象。

Answer 1

Tensorflow对象检测API支持从预先训练的MS COCO检查点开始。刚设定

fine_tune_checkpoint: "/usr/home/username/tmp/model.ckpt-#####"
from_detection_checkpoint: true

在您的检测管道中。（官方参考：https : //github.com/tensorflow/models/blob/master/research/object_detection/g3doc/configuring_jobs.md）

您应该将带有额外类别的图像添加到coco数据集中，并对所有91个类别进行微调，否则您的网络可能会忘记先前学习的内容，而只会检测到新对象。