我正在尝试在tensorflow对象检测API中使用Inception SSD模型。要初始化权重,我想使用预训练的Inception V2在图像网上作为特征提取器。我看到模型配置文件允许你在COCO上使用预训练模型,但如果我想使用Image网络模型,我应该怎么做呢?
答案 0 :(得分:0)
要在Imagenet分类模型上进行训练,请执行以下操作:
1)从预先训练的模型"下载预先训练的模型。 Slim page部分
2)将fine_tune_checkpoint
指向该目录
3)将from_detection_checkpoint
设置为false(因为您现在将从分类检查点进行微调)
请注意,从Imagenet分类检查点进行培训需要更多时间。