tensorflow - 重新训练 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 个类进行微调，否则您的网络可能会忘记之前学到的内容而只检测新对象。