doc-ai / tensorio-android

TensorIO for Android
2 stars 6 forks source link

Start initial work on full TensorFlow integration #181

Closed phildow closed 3 years ago

phildow commented 4 years ago

First Build: 1.13 with Inference

As far as I can tell, last build officially supplied by the TensorFlow team. Use:

implementation 'org.tensorflow:tensorflow-android:1.13.1'

Trying to load a model. Need to update JSON parsing and the layer interface to support batching.

Second Build: 1.13 with Training

Unsure if this is possible yet but see the org.tensorflow.* package. It looks like it contains all the objects needed for this kind of thing. But we may not have full op support for the classification models we test against and almost certainly not for our own models.

Third Build: 1.15 | 2.0 with Inference/Training

Custom build from our doc.ai fork.

2.0 does have a build_all_android script in tensorflow/contrib/makefile and there are other instructions around the net for building tensorflow via bazel. JNI and NDK are required.

phildow commented 4 years ago

Instructions:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/java/README.md https://www.tensorflow.org/install/lang_java https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/android/inference_interface