Tensorflow对象检测API未经训练的Faster-RCNN模型

时间:2019-07-31 12:19:26

标签: python tensorflow object-detection object-detection-api faster-rcnn

我目前正在尝试使用Tensorflow对象检测API和python构建对象检测器。我已经按照herehere

发布的说明设法对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

1 个答案:

答案 0 :(得分:0)

要从头开始训练自己的模型,请执行以下操作:

  1. 注释以下几行
    # fine_tune_checkpoint: <YOUR PATH>
    # from_detection_checkpoint: true
  1. 删除已下载的预训练模型或重命名其路径,以防您遵循本教程。

您不必下载“空”模型。相反,您可以在配置文件中指定自己的体重初始化,例如,如此处所示:How to initialize weight for convolution layers in Tensorflow Object Detection API?