代码之家 › 专栏 › 技术社区 › Michael

在移动ios上运行更快的rcnn

coremltools tensorflow-lite object-detection-api coreml tensorflow

1

Michael · 技术社区 · 7 年前

我有一个更快的rcnn模型,我用gpu(用google models api训练)训练和处理我的google云实例,
我想在mobile上运行它,我找到了一些github,展示了如何运行ssdmobilenet,但是我找不到一个运行得更快的rcnn。
我现在不关心实时性。
我有iphone 6,ios 11.4
该模型可用于金属、coreml、tensorflow-lite… 但对于poc,我需要它在移动设备上运行,而不需要训练新的网络。
有什么帮助吗?
谢谢!

1 回复 | 直到 7 年前

1

Matthijs Hollemans 7 年前

更快的r-cnn需要一些定制层,这些层在metal、coreml等中是不可用的。您必须自己实现这些定制层(或者请人为您实现它们,wink wink)。

我不确定TF Lite是否能工作。它只支持有限数量的ios操作,所以它可能没有r-cnn所需要的一切。但那将是第一件尝试的事。如果这不起作用,我会尝试一个带有自定义层的核心ml模型。

请参阅此处有关Core ML中自定义层的信息: http://machinethink.net/blog/coreml-custom-layers/