在移动iOS上运行Faster-rcnn

时间:2018-06-14 11:30:26

标签: tensorflow coreml object-detection-api tensorflow-lite coremltools

我有更快的rcnn模型,我训练并使用GPU(使用谷歌模型API训练)在我的谷歌云实例上工作,
我想在移动设备上运行它,我发现一些GitHub显示了如何运行SSDmobileNet但我找不到运行Faster-rcnn的那个。 实时不是我现在关注的问题 我有iPhone 6,iOS 11.4
该模型可以使用Metal,CoreML,tensorflow-lite运行...... 但是对于POC我需要它在没有火车新网络的情况下在移动设备上运行 任何帮助?
谢谢!

1 个答案:

答案 0 :(得分:1)

更快的R-CNN需要许多自定义图层,这些图层在Metal,CoreML等中不可用。您必须自己实现这些自定义图层(或雇用某人为您实现它们,眨眼间眨眼)。

我不确定TF-lite是否有效。它只支持在iOS上进行有限数量的操作,因此很有可能它不会拥有更快的R-CNN所需的一切。但这将是第一个尝试的事情。如果这不起作用,我会尝试使用自定义图层的Core ML模型。

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