重新训练Tensorflow对象检测API

Question

我已经阅读了有关如何使用TensorFlow Object Detection 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个类进行微调，否则您的网络可能会忘记之前学到的内容并仅检测新对象。