Skip to content

Instantly share code, notes, and snippets.

@thepulkitagarwal

thepulkitagarwal/TensorflowAndroidSample.java

Forked from YoniTsafir/TensorflowAndroidSample.java
Last active July 31, 2019 11:04
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save thepulkitagarwal/4dc64626470d535edda7c7dfa77636fb to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save thepulkitagarwal/4dc64626470d535edda7c7dfa77636fb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
TensorflowAndroidSample.java
import org.tensorflow.contrib.android.TensorFlowInferenceInterface;
/** One time initialization: */
TensorFlowInferenceInterface tensorflow = new TensorFlowInferenceInterface();
tensorflow.initializeTensorFlow(getAssets(), "file:///android_asset/model.pb");
/** Continuous inference (floats used in example, can be any primitive): */
// loading new input
tensorflow.fillNodeFloat("input:0", INPUT_SHAPE, input); // INPUT_SHAPE is an int[] of expected shape, input is a float[] with the input data
// running inference for given input and reading output
String outputNode = "output:0";
String[] outputNodes = {outputNode};
tensorflow.runInference(outputNodes);
tensorflow.readNodeFloat(outputNode, output); // output is a preallocated float[] in the size of the expected output vector
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment