我目前正在尝试使用Tensorflow对象检测API和python构建对象检测器。我已经按照here和here
发布的说明设法对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
答案 0 :(得分:0)
要从头开始训练自己的模型,请执行以下操作:
# fine_tune_checkpoint: <YOUR PATH>
# from_detection_checkpoint: true
您不必下载“空”模型。相反,您可以在配置文件中指定自己的体重初始化,例如,如此处所示:How to initialize weight for convolution layers in Tensorflow Object Detection API?