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

Dux*_*cis 1 neural-network ios tensorflow coreml

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

Bra*_*son 5

支持在刚刚宣布导入TensorFlow型号为核心ML.这是使用tfcoreml转换器完成,它应该接收.pb图并输出Core ML模型.从那里,您可以将此模型与Core ML一起使用,并接收静止图像或视频帧进行处理.

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