Tensorflow 对象检测 API 未经训练的 Faster-RCNN 模型
Tensorflow Object Detection API Untrained Faster-RCNN Model
我目前正在尝试使用 Tensorflow 对象检测 API 和 python 构建一个对象检测器。我已经按照 here and here
发布的说明成功地重新训练了 faster-rcnn model
但是据我所知,训练时间相当长。我知道我正在使用迁移学习,而不是从头开始训练 faster-rcnn 模型。我想知道是否可以下载未经训练的 faster-rcnn 模型并从头开始(端到端)训练它,而不必求助于迁移学习。
我熟悉迁移学习的优点,但是,我的对象检测器的目标是快速可训练,范围狭窄,并且在字母而不是对象上进行训练,所以我认为迁移学习不是最好的路线。
我相信解决这个问题与 pipeline.config
文件有关,特别是在这部分:
fine_tune_checkpoint: "PATH/TO/PRETRAINED/model.ckpt"
from_detection_checkpoint: true
num_steps: 200000
但是我不知道如何指定没有fine_tune_checkpoint
要从头开始训练您自己的模型,请执行以下操作:
- 注释掉以下行
# fine_tune_checkpoint: <YOUR PATH>
# from_detection_checkpoint: true
- 删除您下载的预训练模型或重命名其路径,以防您按照教程进行操作。
您不必下载 "empty" 模型。相反,您可以在配置文件中指定您自己的权重初始化,例如,如下所示:
我目前正在尝试使用 Tensorflow 对象检测 API 和 python 构建一个对象检测器。我已经按照 here and here
发布的说明成功地重新训练了faster-rcnn model
但是据我所知,训练时间相当长。我知道我正在使用迁移学习,而不是从头开始训练 faster-rcnn 模型。我想知道是否可以下载未经训练的 faster-rcnn 模型并从头开始(端到端)训练它,而不必求助于迁移学习。
我熟悉迁移学习的优点,但是,我的对象检测器的目标是快速可训练,范围狭窄,并且在字母而不是对象上进行训练,所以我认为迁移学习不是最好的路线。
我相信解决这个问题与 pipeline.config
文件有关,特别是在这部分:
fine_tune_checkpoint: "PATH/TO/PRETRAINED/model.ckpt"
from_detection_checkpoint: true
num_steps: 200000
但是我不知道如何指定没有fine_tune_checkpoint
要从头开始训练您自己的模型,请执行以下操作:
- 注释掉以下行
# fine_tune_checkpoint: <YOUR PATH>
# from_detection_checkpoint: true
- 删除您下载的预训练模型或重命名其路径,以防您按照教程进行操作。
您不必下载 "empty" 模型。相反,您可以在配置文件中指定您自己的权重初始化,例如,如下所示: