是否可以在 Tensorflow 的对象检测 API 中使用具有 SSD 元架构的不同特征提取器？我知道提供了用于 mobilenets 和 inception 的 .config 文件，但是否可以使用不同的架构，如 AlexNet 或 VGG？
问问题
1241 次
1 回答
3
这是可能的，但需要做一些工作，如此处所述，您应该阅读此页面以获取详细说明和示例链接。
简而言之，您需要创建一个自定义FasterRCNNFeatureExtractor类，对应于 VGG 或 AlexNet（它可能需要一些关于这些的知识，例如涉及的子采样量）。在本课程中，您将编写如何预处理数据、如何检索其中的第一和第二阶段特征（通常如何调用最后一个卷积层）以及如何加载数据的代码。
然后你需要通过修改文件来注册你的特征提取器（告诉对象检测 API 它存在）object_detection/builders/model_builder.py。
最后，您应该能够使用自定义特征提取器制作配置文件，等等！
于 2018-01-23T16:31:17.027 回答