Skip to content

Latest commit

 

History

History
43 lines (31 loc) · 2.72 KB

File metadata and controls

43 lines (31 loc) · 2.72 KB

EdgeTpu SSDLite Mobiledet Transfer Learning Turotial

License made-with-coral

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.

Open In Colab

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

Open In Colab

Eval:

References: