是否可以在Tensorflow的Object Detection API中使用具有SSD元架构的不同特征提取器?我知道提供了针对mobilenet和inception的.config文件,但是可以使用像AlexNet或VGG这样的其他架构吗?
答案 0 :(得分:3)
这是可能的,但正如here所解释的那样,您应该阅读此页面以获取详细说明和示例链接。
简而言之,您需要创建一个自定义FasterRCNNFeatureExtractor
类,对应于VGG或AlexNet(可能需要一些关于这些的知识,例如调用的子采样量)。在本课程中,您将编码应如何预处理数据,如何检索其中的第一和第二阶段特征(通常如何调用最后一个卷积层)以及如何加载它。
然后,您需要通过修改文件object_detection/builders/model_builder.py
来注册您的特征提取器(告诉对象检测API它存在)。
最后,您应该可以使用自定义功能提取器制作配置文件,等等!