Pocket AutoML: train a deep learning model on a phone, follow the tutorial to create an app with it

OutSorcerer

New member
Aug 16, 2021
1
1
0
Visit site
Hello everyone!

I made an Android app called Pocket AutoML that trains a deep learning model for image classification right on your phone and made a tutorial how to export a model from it and make your custom Android app with an exported model like e.g. an app for identifying plants or sorting lego bricks.

Some of its features:
  • Pocket AutoML lets AI enthusiasts even without any prior machine learning expertise train a deep learning image classification model, export it in TensorFlow Lite format and make a custom Android app based on it following the provided tutorial.
  • Computer vision or deep learning professionals can also use it as a tool to create a quick proof-of-concept for transfer learning on their tasks without a single line of code.
  • It trains a model right on on your device in seconds (for a dataset with dozens of images).
  • It respects your privacy: your images are never uploaded anywhere as both training and prediction happens on your device (the apps made with exported models also have the same advantages).
  • It does not need an internet connection for training and predicting (internet connection is needed for TL Lite export though).
  • Just few images per class can be enough to train a model that accurately classifies objects (what is known as few-shot learning).

Pocket AutoML does nothing magical, it just uses transfer learning, which you can use directly, as described in the end of the tutorial above, so no vendor lock is imposed.

The above mentioned app creation tutorial on GitHub includes a working example Android app that classifies images from a phone camera and steps to customize that app with your model trained in Pocket AutoML.

Since Pocket AutoML exports models in TensorFlow Lite format, they can also be used to create apps for platforms other then Android like iOS, embedded Linux devices like Raspberry Pi or Coral and microcontrollers.

You can compare this app with other no-code or low-code deep learning solutions: Create ML from Apple, Lobe from Microsoft, Teachable Machine from Google, Google AutoML Vision, Azure Custom Vision from Microsoft, TensorFlow Lite Model Maker from Google, they are either free like Pocket AutoML or have a trial period.

I will be glad to discuss the app here and help in case of potential technical issues with the app itself or when following the tutorial above.

Links


The screenshots are attached.
 

Attachments

  • screen1.jpg
    screen1.jpg
    87.3 KB · Views: 2
  • screen2.jpg
    screen2.jpg
    413.9 KB · Views: 2
  • screen3.jpg
    screen3.jpg
    290.1 KB · Views: 1
  • screen4.jpg
    screen4.jpg
    177.2 KB · Views: 1
  • screen5.jpg
    screen5.jpg
    184 KB · Views: 1
  • screen6.jpg
    screen6.jpg
    168.8 KB · Views: 1
  • Like
Reactions: alexsaunders790

Forum statistics

Threads
943,007
Messages
6,916,861
Members
3,158,771
Latest member
planeboy