无法运行TensorFlow推理

时间:2019-01-18 16:05:41

标签: android tensorflow

我在android上运行训练有素的yolov2网络时遇到问题。我正在使用该项目进行测试https://github.com/szaza/android-yolo-v2

提供的网络运行正常。问题仅存在于我训练有素的网络中。我使用darkflow将yolov2.weights文件转换为pb文件。我试图找到有关我所遇到的错误类型的信息,但是我对TensorFlow的内部工作原理的了解太少,无法了解正在发生的事情。

看起来冻结的图已正确加载:

I/TensorFlowInferenceInterface: Successfully loaded model from 'file:///android_asset/yolov2_finderpatterns.pb'

所以我想TensorFlow还有其他问题,也许有人可以向我解释此错误的含义:

E/TensorFlowInferenceInterface: Failed to run TensorFlow inference with inputs:[input], outputs:[output]
E/AndroidRuntime: FATAL EXCEPTION: inference
Process: org.tensorflow.yolo, PID: 16974
java.lang.IllegalArgumentException: No OpKernel was registered to support Op 'ExtractImagePatches' with these attrs.  Registered devices: [CPU], Registered kernels:
  <no registered kernels>

     [[{{node ExtractImagePatches}} = ExtractImagePatches[T=DT_FLOAT, ksizes=[1, 2, 2, 1], padding="VALID", rates=[1, 1, 1, 1], strides=[1, 2, 2, 1]](47-leaky)]]

1 个答案:

答案 0 :(得分:1)

tensorflow存储库中存在问题,并有解决方案,请参阅this issue