0

我目前正在尝试使用带有 python 的 Tensorflow 对象检测 API 构建一个对象检测器。我已按照此处和此处faster-rcnn model发布的说明设法重新培训

但是，据我所知，培训时间相当长。我知道我使用的是迁移学习，而不是从头开始训练更快的 rcnn 模型。我想知道是否有办法下载未经训练的 fast-rcnn 模型并从头开始（端到端）进行训练，而不必求助于迁移学习。

我熟悉迁移学习的优点，但是，我的对象检测器旨在快速训练、范围狭窄，并且训练的是字母而不是对象，所以我不认为迁移学习是最好的途径。

我相信解决这个问题将与pipeline.config文件有关，特别是在这部分：

fine_tune_checkpoint: "PATH/TO/PRETRAINED/model.ckpt"
from_detection_checkpoint: true
num_steps: 200000

但我不确定如何指定没有fine_tune_checkpoint

4

1 回答 1

0

要从头开始训练您自己的模型，请执行以下操作：

  1. 注释掉以下几行
    # fine_tune_checkpoint: <YOUR PATH>
    # from_detection_checkpoint: true
  1. 如果您按照教程进行操作，请删除您下载的预训练模型或重命名其路径。

您不必下载“空”模型。相反，您可以在配置文件中指定自己的权重初始化，例如，如下所示：如何在 Tensorflow 对象检测 API 中初始化卷积层的权重？

于 2019-08-30T12:31:11.480 回答