This repo contains the code used in my experiment titled "Detecting Pikachu on Android using Tensorflow Object Detection". In this experiment, which is available here, I explained the many steps needed to train a custom object detection model using TensorFlow Object Detection API and how to deploy it in an Android device.
The content of this repo is mostly divided in 3 parts
- The directory android contains the 'gradle.build' file used to build the example TensorFlow provides, and the file 'DetectorActivity.java' which is the responsible of performing the detection in the app
- The directory training has the final models my training produced, as well as the pipeline configuration file required for the training.
- The rest of the files are those scripts needed to prepare the dataset.
All this code by itself does not do anything. It must be used in combination with the Object Detection API. The report linked above has all the instructions on how to use the code alongside the API.