问如何在tensorflow对象检测api中使用Image net上的预训练模型
EN

Stack Overflow用户

提问于 2017-07-27 04:32:39

回答 1查看 1.6K关注 0票数 0

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

发布于 2017-07-28 09:44:30

要对Imagenet分类模型进行训练，请执行以下操作：

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

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

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/45336863

复制

相似问题

问如何在tensorflow对象检测api中使用Image net上的预训练模型EN