java-如何向Tensorflow Android构建中添加操作

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了java-如何向Tensorflow Android构建中添加操作，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含3555字，纯文字阅读大概需要6分钟。

内容图文

我试图在Android上运行Tensorflow模型,因此我正在使用nightly native build in here,继Android demo之后,我已经成功运行了Tensorflow Android lib,并使用以下代码加载了该模型.

inferenceInterface = new TensorFlowInferenceInterface(getAssets(), MODEL_FILE);

日志显示结果很好.

I/TensorFlowInferenceInterface: Successfully loaded TensorFlow native methods (RunStats error may be ignored)
I/TensorFlowInferenceInterface: Model load took 1007ms, TensorFlow version: 1.2.0-rc0
I/TensorFlowInferenceInterface: Successfully loaded model from 'file:///android_asset/model.pb'

但是,当我完成所有输入节点的输入时

inferenceInterface.feed("input1", new int[]{1, 2, 3}, 1, 3);
inferenceInterface.feed("input2", new int[]{3}, 1);
inferenceInterface.feed("input3", new int[]{4}, 1);

然后调用run方法

inferenceInterface.run(new String[]{"output"});

Tensorflow被打破,说某些内核未注册

E/TensorFlowInferenceInterface: Failed to run TensorFlow inference with inputs:[input1, input2, input3], outputs:[output]
D/AndroidRuntime: Shutting down VM
E/AndroidRuntime: FATAL EXCEPTION: main
Process: ...jnitest, PID: 16357
java.lang.RuntimeException: Unable to start activity ComponentInfo{...MainActivity}: java.lang.IllegalArgumentException: No OpKernel was registered to support Op 'LessEqual' with these attrs.  Registered devices: [CPU], Registered kernels:
<no registered kernels>

[[Node: .../LessEqual = LessEqual[T=DT_INT32](.../maximum_iterations, .../LessEqual/y)]]
at android.app.ActivityThread.performLaunchActivity(ActivityThread.java:2665)
at android.app.ActivityThread.handleLaunchActivity(ActivityThread.java:2726)
at android.app.ActivityThread.-wrap12(ActivityThread.java)
at android.app.ActivityThread$H.handleMessage(ActivityThread.java:1477)
at android.os.Handler.dispatchMessage(Handler.java:102)
at android.os.Looper.loop(Looper.java:154)
at android.app.ActivityThread.main(ActivityThread.java:6119)
at java.lang.reflect.Method.invoke(Native Method)
at com.android.internal.os.ZygoteInit$MethodAndArgsCaller.run(ZygoteInit.java:886)
at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:776)
Caused by: java.lang.IllegalArgumentException: No OpKernel was registered to support Op 'LessEqual' with these attrs.  Registered devices: [CPU], Registered kernels:
<no registered kernels>

[[Node: dynamic_seq2seq/decoder/decoder_1/LessEqual = LessEqual[T=DT_INT32](.../maximum_iterations, .../LessEqual/y)]]
at org.tensorflow.Session.run(Native Method)
at org.tensorflow.Session.access$100(Session.java:48)
at org.tensorflow.Session$Runner.runHelper(Session.java:295)
at org.tensorflow.Session$Runner.run(Session.java:245)
at org.tensorflow.contrib.android.TensorFlowInferenceInterface.run(TensorFlowInferenceInterface.java:142)
at org.tensorflow.contrib.android.TensorFlowInferenceInterface.run(TensorFlowInferenceInterface.java:111)
at ...jnitest.MainActivity.onCreate(MainActivity.java:52)
at android.app.Activity.performCreate(Activity.java:6679)
at android.app.Instrumentation.callActivityOnCreate(Instrumentation.java:1118)
at android.app.ActivityThread.performLaunchActivity(ActivityThread.java:2618)
.. 9 more

我认为int32的“ LessEqual”应在Tensorflow中定义,但不能与Tensorflow Android Lib一起构建.

因此,我的问题是如何在Android lib构建中包含更多内核或以其他任何方式解决此问题,我们将不胜感激.

解决方法:

我建议您阅读此免费电子书：Building Mobile Applications with TensorFlow

它包含有关移动设备上提供哪些操作的部分. (第33页),其中说明了如何为那些默认情况下不用于移动版本的op内核将实现添加到版本中.

仅供参考,这本电子书由Pete Warden(GitHub,blog)撰写,他在Google工作,是TensorFlow的维护者之一.

内容总结

以上是互联网集市为您收集整理的java-如何向Tensorflow Android构建中添加操作全部内容，希望文章能够帮你解决java-如何向Tensorflow Android构建中添加操作所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---