如何在TensorFlow对象检测API中从头开始训练？

Question

github问题2446说如果我们想要从头开始训练，我们需要在<{1}}中评论这一行：

model.config

我遵循了这个问题并评论了这一行，用于在我自己的数据集中训练SSD_mobilenet。 它可以训练，损失可以降低，但准确度保持在0.0 。

我之前使用物体检测api与模型动物园的预训练模型，它在mAP = 90％时效果很好，这两个任务之间的唯一区别是#fine_tune_checkpoint: "PATH_TO_BE_CONFIGURED/model.ckpt" from_detection_checkpoint: true 中的注释行。

我想知道如何使用TensorFlow中的对象检测api从头开始训练检测模型。谢谢！

Answer 1

我建议你继续微调你的模型，而不是从头开始训练。从头开始训练模型需要更多的计算时间来使模型收敛，这意味着你再次从随机权重开始，你正在调整它们直到它们收敛，这将最终收敛，但收敛的时间可能是一周或一个月因为它是一个大型神经网络。

在微调中，您将使用之前的权重，但您将根据自己的数据集对其进行调整，最终结果是您的模型可帮助您在自己的数据集中进行检测。最终模型也或多或少是从头开始训练的结果，因为模型只适用于您的新数据集和新类。