gpt4 book ai didi

android - 无法运行 TensorFlow 推理

转载 作者:行者123 更新时间:2023-11-29 23:21:21 31 4
gpt4 key购买 nike

我在 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)]]

最佳答案

tensorflow repository 有问题,解决方法请看this issue

关于android - 无法运行 TensorFlow 推理,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54257552/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com