1

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

4

1 回答 1

1

Faster R-CNN 需要一些在 Metal、CoreML 等中不可用的自定义层。您必须自己实现这些自定义层(或雇人为您实现它们,眨眼)。

我不确定 TF-lite 是否可以工作。它仅支持 iOS 上有限数量的操作,因此它可能不会拥有 Faster R-CNN 所需的一切。但这将是首先要尝试的事情。如果这不起作用,我会尝试使用自定义层的 Core ML 模型。

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

于 2018-06-14T12:33:21.237 回答