跑隐含模型,其预训练模型并没有在官网公布,需要重头开始训练
步骤:
1、复制同类配置文件,修改其中的特征提取网络名称,并注释掉加载预训练权重的代码
2、在python安装目录下的object_detection下,修改builders/model_builder.py
(我的目录在/home/ubuntu/anaconda3/lib/python3.6/site-packages/object_detection-0.1-py3.6.egg/object_detection/builders/model_builder.py),添加类似如下两项即可,我这里添加的是faster_rcnn_mobilenet_v1模型,其他模型类似: