如何在张量流对象检测api中使用Image Web上的预训练模型

时间:2017-07-26 20:32:39

标签: tensorflow object-detection

我正在尝试在tensorflow对象检测API中使用Inception SSD模型。要初始化权重,我想使用预训练的Inception V2在图像网上作为特征提取器。我看到模型配置文件允许你在COCO上使用预训练模型,但如果我想使用Image网络模型,我应该怎么做呢?

1 个答案:

答案 0 :(得分:0)

要在Imagenet分类模型上进行训练,请执行以下操作:

1)从预先训练的模型"下载预先训练的模型。 Slim page部分 2)将fine_tune_checkpoint指向该目录 3)将from_detection_checkpoint设置为false(因为您现在将从分类检查点进行微调)

请注意,从Imagenet分类检查点进行培训需要更多时间。