如何在iOS中运行Tensorflow Object Detection API模型？

Question

我刚训练了一个结果令人满意的模型，我有了frozen_inference_graph.pb。我将如何在iOS上运行此操作？如果有帮助，它在SSD Mobilenet V1上接受过培训。最理想的是我想用GPU运行它（我知道在iOS上不能做到张量流API），但是首先在CPU上运行它会很棒。

Answer 1

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

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