从对象检测API模型中提取研磨锚

时间:2018-11-14 19:48:11

标签: tensorflow tensor coreml object-detection-api

我目前正在尝试使我的SSDLite网络与iOS一起使用Tensroflow对象检测API进行培训。

所以我正在使用SSDMobileNet_CoreML的开源代码。

该图已经可以使用,但有一些限制。为了在iOS上运行,我必须从Graph中提取FeatureExtractor,并且无法保留Preprocessor,Posprocessor和MutlipleGrindAnchorBox,就像在SSDMobileNet_CoreML中所做的一样。

在这里您可以看到他们使用的Anchors

因此,由于我的锚似乎有些不同,所以我试图不理解他们如何获得此阵列。

因此,我在GitHub Issue上发现了一个有趣的例子,创建Anchors的用户解释了如何获得它们。

他说:

  

我刚刚从import / MultipleGridAnchorGenerator / Identity张量中将它们从Tensorflow图中导出

我已经在Graph中找到了匹配的张量,但是我不知道如何导出Graph并检索正确的Anchor编码。 有人可以向我解释吗?

1 个答案:

答案 0 :(得分:0)

我已经知道了。在引号下方有一个指向Python Notebook的链接,该链接详细解释了所有内容。