在Tensorflows的对象检测API

时间:2018-01-23 16:08:59

标签: tensorflow object-detection object-detection-api

是否可以在Tensorflow的Object Detection API中使用具有SSD元架构的不同特征提取器?我知道提供了针对mobilenet和inception的.config文件,但是可以使用像AlexNet或VGG这样的其他架构吗?

1 个答案:

答案 0 :(得分:3)

这是可能的,但正如here所解释的那样,您应该阅读此页面以获取详细说明和示例链接。

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

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

最后,您应该可以使用自定义功能提取器制作配置文件,等等!