EdgeTpu SSDLite Mobiledet Transfer Learning Turotial

This repo contains the Colab notebook for my ssdlite mobiledet model retrain tutorial as well as the minimal example code to deploys the model on the EdgeTpu.

Checkout my Medium article for the full walk through:

Actual eval after about 3k steps:

Quick run:

$ python3 run_model.py models/ssdlite_mobiledet_dog_vs_cat_edgetpu.tflite test_images
Evaluating: test_images/image1.jpg
Evaluating: test_images/image2.jpg
Evaluating: test_images/image3.jpg
Evaluating: test_images/image4.jpg
Evaluating: test_images/image5.jpg
Evaluating: test_images/image6.jpg
Evaluating: test_images/image7.jpg
Evaluating: test_images/image8.jpg
Inference time:  0.06407481100177392

Inference Results

[Update]: Added new colab for Mask Detection

Eval:

References:

• Yunyang Xiong, Hanxiao Liu, Suyog Gupta, Berkin Akin, Gabriel Bender, Pieter-Jan Kindermans, Mingxing Tan, Vikas Singh, Bo Chen: MobileDets: Searching for Object Detection Architectures for Mobile Accelerators. CoRR abs/2004.14525 (2020)