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

Question

我刚刚训练了一个模型，并获得了令人满意的结果，并且得到了Frozen_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结果以获取对象类和边界框。