如何在张量流对象检测API中使用图像网上的预训练模型

Question

我想在张量流对象检测API中使用初始SSD模型。为了初始化权重，我想使用pretrained Inception V2在图像网上作为特征提取器。我看到模型配置文件允许您在COCO上使用预训练模型，但是如果我想使用图像网络模型，我应该如何处理它？

Answer 1

要在Imagenet分类模型训练，做到以下几点：

1）从“预训练的模型”部分下载预先训练模型的Slim page 2）点fine_tune_checkpoint在该目录 3 ）将from_detection_checkpoint设置为假（因为您现在将从分类检查点进行微调）

请注意，从Imagenet分类检查点进行训练将需要大量更多时间。