TensorFlow - 为iOS导出对象检测模型

时间:2017-08-04 20:14:18

标签: tensorflow object-detection

我使用对象检测API训练了对象检测模型,需要将其导出到iOS上运行。 我使用export_inference_graph.py创建了一个.pb文件,但不幸的是,这个文件在iOS应用程序(https://github.com/KleinYuan/tensorflow-yolo-ios)中使用时产生了加载错误。 我已经读过,还有其他步骤可以冻结然后简化模型,但我还没有找到一个逐步的解决方案来做到这一点。有什么建议吗?

错误: yolo-ios / tensorflow_utils.mm:169]无法创建TensorFlow图:无效参数:没有注册OpKernel以支持使用这些attrs的Op'Equal'。已注册的设备:[CPU],已注册的内核:   设备= 'CPU'; T [DT_FLOAT]

 [[Node: Postprocessor/BatchMultiClassNonMaxSuppression/MultiClassNonMaxSuppression/Equal = Equal[T=DT_INT32](Postprocessor/BatchMultiClassNonMaxSuppression/MultiClassNonMaxSuppression/strided_slice, Postprocessor/BatchMultiClassNonMaxSuppression/MultiClassNonMaxSuppression/strided_slice_1)]]

ps:请注意,如果我在ipython笔记本https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb中使用它,训练好的模型在PC上加载并运行正常<{3}}

0 个答案:

没有答案