在Tensorflow的对象检测API中，特征提取器与ssd元架构的不同用法

Question

在Tensorflow的对象检测API中，是否可以将不同的特征提取器与SSD元架构一起使用？我知道已经提供了用于mobilenets和inception的.config文件，但是是否可以使用不同的架构，如AlexNet或VGG？

Answer 1

这是可能的，但需要做一点工作，正如here所解释的那样，您应该阅读此页面以获得详细的解释和示例链接。

简而言之，您将需要创建一个与VGG或AlexNet相对应的自定义FasterRCNNFeatureExtractor类(这可能需要一些关于它们的知识，例如所涉及的子采样量)。在这个类中，你将编码你的数据应该如何预处理，如何检索其中的第一和第二阶段特征(通常是如何调用最后一个卷积层)，以及如何加载它。

然后，您需要通过修改文件object_detection/builders/model_builder.py来注册您的羽毛提取器(告诉对象检测API它是存在的)。

最后，你应该能够用你的自定义特征提取器来创建一个配置文件了！