如何在 iOS 中运行 Tensorflow 对象检测 API 模型？

Question

我刚刚训练了一个结果令人满意的模型，并且我有 freeze_inference_graph.pb 。我将如何在 iOS 上运行它？如果有帮助，它在 SSD Mobilenet V1 上进行了培训。理想情况下，我想使用 GPU 运行它（我知道 tensorflow API 在 iOS 上无法做到这一点），但最好先在 CPU 上运行它。

Answer 1

支持 just announced 将 TensorFlow 模型导入 Core ML。这是使用the tfcoreml converter 完成的，它应该接收您的.pb 图形并输出Core ML 模型。从那里，您可以将此模型与 Core ML 一起使用，并获取静止图像或视频帧进行处理。

此时，您需要确保提供正确的输入色彩空间和尺寸，然后正确提取和处理 SSD 结果，以获得您的对象类和边界框。