Skip to main content

Training a Custom Classifier Model

Google's Teachable Machine is a pure browser-based tool that allows users to create classifier models without any code or advanced understanding of machine learning.

Data Collection

Gather images for each class you want to recognize. Ensure that you have a balanced dataset (a roughly equal number of images for each class). Ensure you have diversity in camera angle, lighting, backgrounds, object characteristics, etc.

Training with Teachable Machine

Go to the Teachable Machine website: teachable machine

  • Click on the “Get Started” button.
  • Select the “Image Project” to create an image classifier.
  • Select "Standard image model"
  • For each class:
    • Click on the 'Add Class' button.
    • Upload the images pertaining to that class.
  • After adding all classes, click on the “Train Model” button.
  • Once training is complete, you can test the model right in the browser.
  • Click the "Export Model" Button
  • Navigate to the "Tensorflow Lite" tab
  • Select "EdgeTPU"
  • Download the model.