如何在Android上使用JNI训练张量流网络?

时间:2016-10-18 06:38:36

标签: android neural-network tensorflow

Tensorflow旨在与移动设备兼容,there is an example of android app使用tensorflow。但我无法在源代码中找到任何与神经网络相关的函数。 在tensorflow C++ API中,我无法找到任何用于培训网络的API?

我是不是太粗心无法找到它们或者还有其他方法可以实现这个目标?

感谢。

1 个答案:

答案 0 :(得分:0)

Android并不是Tensorflow的培训框架。 目前在已发布的(唯一的“开始”)演示中,您只能这样做 用它来推断。

如果你坚持使用这种基于Arm的设备进行训练,你将需要使用所有训练操作编译TF for android。在优化推理模型(用于移动)的过程中通常省略的内核,以及编写用于训练c ++ api的JNI接口的内核。

更好的方法是使用所谓的“TF服务”并使用RPC进行云培训。您可以在手机上收集数据,然后将其发送给培训或微调预先训练的模型,然后将微调模型返回到您的手机。看这个演示: Tensorflow Cloud Demo