This Android application is used to test inferences of Tensorflow/Pytorch graphs. This application supports variety of Models, Computer Vision tasks independent of library. Also one can use it to learn how the code works for pytorch and tensorflow. Soon more tasks will be added gradually.
- Fast inference on android for tensorflow/pytorch graphs.
- Supports Classification, Object Detection(Faster R-CNN).
- Simple GUI based options for inputs.
- Auto detect library(Tensorflow/Pytorch).
- Small size and easy to understand code(Tons of comments).
- Open source: Edit/Make changes, use as a starter template No Worries.
- Train your model.
- Save it as .pt for pytorch or .tflite for tensorflow.
- Put labels(classes names) into a simple .txt file, no formatting needed only newline for new class. Same for object detection.
- Select files through application setting.
- Select image and press button.
- Done.
- More Image tasks.
You can compile above code into Android studio or download apk
1.0 here,
Update 1.1 here
Classification Models Zoo:
Pytorch (https://github.com/qubvel/classification_models)
Tensorflow (https://www.tensorflow.org/lite/guide/hosted_models)
Object Detection Models Zoo: (https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md)
- Changed code to completely follow the official documentations.
- Added support for camerax.
- Tensorflow runs on 0.0.0-nightly version.
- Supports Quant/Float models on Tensorflow.