To use this app, you need to have a pre-trained image classification TF Lite model. This app can be used for any custom-trained model.
It is assumed that you already have a TF Lite image classification model of yourself.
-
Clone the repository on your local machine.
-
Copy your
.tflitemodel and its correspondinglabel.txtfile insideCartoon-Classification-on-Android-using-TF-Lite\app\src\main\assetsfolder. -
Open the project in Android Studio and let the project build itself.
-
Open
MainActivity.javafile and edit Line 169 by replacingcartoon_model.tflitewith the name of your TF Lite model. Also edit Line 231 by replacingcartoon_labels.txtwith the name of your model's label file. -
Build and run the app on your physical device or on emulator.
- Read the Medium blog for step-by-step implementation.
