How to Train TensorFlow Lite Object Detection Models Using Google Colab | SSD MobileNet

Edje Electronics β€’ February 13, 2023
Video Thumbnail
Edje Electronics Logo

Edje Electronics

@edjeelectronics

About

Looking for fun projects to do with your Raspberry Pi? You've come to the right place! I post videos about software and hardware related projects that involve the Pi. Currently, I'm working on creating a blackjack-playing robot that uses deep learning neural networks to identify playing cards, so most of my videos are on the topic of computer vision. You can view code for the projects I've worked on at https://github.com/EdjeElectronics. Want help developing vision-capable products for your business? My company, EJ Technology Consultants, can help! Learn more here: https://ejtech.io As an Amazon Associate, I earn from qualifying purchases through links that are posted on my video descriptions.

Video Description

Let's train, export, and deploy a TensorFlow Lite object detection model on the Raspberry Pi - all through a web browser using Google Colab! We'll walk through a Colab notebook that provides start-to-finish code and instructions for training a custom TFLite model, and then show how to run it on a Raspberry Pi. The notebook uses the TensorFlow Object Detection API to train SSD-MobileNet or EfficientDet models and converts them to TFLite format. Click this link to the Colab notebook to get started: https://colab.research.google.com/github/EdjeElectronics/TensorFlow-Lite-Object-Detection-on-Android-and-Raspberry-Pi/blob/master/Train_TFLite2_Object_Detction_Model.ipynb *WARNING:* Google deprecated the TensorFlow Object Detection API over two years ago. For the sake of legacy code, I've kept this training notebook on life support through various hacks and band-aid fixes, and it is prone to stop working at any point. I will not be providing further support for this video or training notebook. I highly recommend using the newer PyTorch-based Ultralytics YOLO models for object detection. They perform better and they're much easier to work with. See my video tutorial on how to train YOLO detection models here. https://youtu.be/r0RspiLG260 -- Other Links -- πŸ“Έ How to capture and label training data for object detection models: https://youtu.be/v0ssiOY6cfg πŸ… TFLite model comparison article: https://ejtech.io/learn/tflite-object-detection-model-comparison πŸ“ Instructions to set up TFLite on the Raspberry Pi: https://www.youtube.com/watch?v=aimSGOAUI8Y πŸ’» Instructions to run TFLite models on Windows: https://github.com/EdjeElectronics/TensorFlow-Lite-Object-Detection-on-Android-and-Raspberry-Pi/blob/master/deploy_guides/Windows_TFLite_Guide.md 🐜 How to quantize your TFLite model: Still to come! πŸ“„ TFLite GitHub repository: https://github.com/EdjeElectronics/TensorFlow-Lite-Object-Detection-on-Android-and-Raspberry-Pi -- Chapters -- 0:00 Introduction 1:06 Google Colab 1:41 1. Gather Training Images 3:22 2. Install TensorFlow 4:43 3. Upload Images and Prepare Data 8:41 4. Set up Training Configuration 11:20 5. Train Model 13:48 6. Convert Model to TFLite 14:20 7. Test Model 17:50 8. Deploy Model 22:07 9. Quantization 22:30 Conclusion -- Music -- - Blue Wednesday – Japanese Garden - Provided by Lofi Records - Watch: https://youtu.be/vJ0Sty6K2cU - Download/Stream: https://fanlink.to/Discovery

Essential TensorFlow Tools

AI-recommended products based on this video

Loading...
PowerA Protection Case for Nintendo Switch - OLED Model, Nintendo Switch, Nintendo Switch Lite - Fortnite Peely, Officially licensed, Bonus Virtual Item Included.

PowerA Protection Case for Nintendo Switch - OLED Model, Nintendo Switch, Nintendo Switch Lite - Fortnite Peely, Officially licensed, Bonus Virtual Item Included.

(488)
$23.53
FREE delivery Sun, Jun 15 on your first order
Loading...
PowerA Protection Case for Nintendo Switch - OLED Model, Nintendo Switch and Nintendo Switch Lite - Sonic Peel Out

PowerA Protection Case for Nintendo Switch - OLED Model, Nintendo Switch and Nintendo Switch Lite - Sonic Peel Out

(195)
$25.49
FREE delivery Sun, Jun 15 on your first order
Loading...
PowerA Protection Case for Nintendo Switch - OLED Model, Nintendo Switch or Switch Nintendo Lite - Rainbow Run Mario - Protection Case Edition

PowerA Protection Case for Nintendo Switch - OLED Model, Nintendo Switch or Switch Nintendo Lite - Rainbow Run Mario - Protection Case Edition

(1,309)
$25.19
FREE delivery Sun, Jun 15 on your first order
Loading...
Asus Vivobook 14" FHD Laptop - Intel Core i7-1355U, 12GB RAM, 512GB SSD, Windows 11, Newest Model - Quite Blue (X1404VA-I712512)

Asus Vivobook 14" FHD Laptop - Intel Core i7-1355U, 12GB RAM, 512GB SSD, Windows 11, Newest Model - Quite Blue (X1404VA-I712512)

(3)
$749.99$620.62
FREE delivery Sat, Jan 3
Loading...
Asus Vivobook Go 11.6" HD Slim Laptop - Intel Celeron N4500, 4GB RAM, 128GB SSD, Windows 11, Newest Model (with Microfiber Cloth) - Black (L210KA-ES04)

Asus Vivobook Go 11.6" HD Slim Laptop - Intel Celeron N4500, 4GB RAM, 128GB SSD, Windows 11, Newest Model (with Microfiber Cloth) - Black (L210KA-ES04)

(0)
$239.99
FREE delivery Sat, Jan 3
Loading...
LK 6 Pack Screen Protector for Apple Watch Series 11 46mm, Self-Healing TPU Material for iWatch 46mm Screen Protector, Scrathes-Resistant, Bubble-Free, HD Clarity, Anti-Glare, Transparent Global Recycled Standard
Best Seller

LK 6 Pack Screen Protector for Apple Watch Series 11 46mm, Self-Healing TPU Material for iWatch 46mm Screen Protector, Scrathes-Resistant, Bubble-Free, HD Clarity, Anti-Glare, Transparent Global Recycled Standard

(1,111)
$16.99
FREE delivery Sun, Oct 5 on your first order
Loading...
ivoler Dockable Case for Nintendo Switch 2 (2025), Slim Carrying Protective PC Case, Soft TPU Grip Case for Joy-Con, Grip Cover with Shock-Absorption, Anti-Scratch, Split Design- Black

ivoler Dockable Case for Nintendo Switch 2 (2025), Slim Carrying Protective PC Case, Soft TPU Grip Case for Joy-Con, Grip Cover with Shock-Absorption, Anti-Scratch, Split Design- Black

(1,206)
$25.64
FREE delivery Sun, Jun 15 on your first order
50+ bought in past month
Loading...
HEYSTOP Switch Case for Nintendo Clear TPU Grip Protective Cover Case Compatible Accessories, Shock-Absorption Design, with a Tempered Glass Screen Protector and Thumb Caps

HEYSTOP Switch Case for Nintendo Clear TPU Grip Protective Cover Case Compatible Accessories, Shock-Absorption Design, with a Tempered Glass Screen Protector and Thumb Caps

(1,741)
$22.99
FREE delivery Sun, Jun 15 on your first order
Loading...
OHLPRO Tablet Holder for Car, iPad Holder for Car Air Vent Mount, Universal Dashboard Windshield 2-in-1 Cradle TPU Suction Sticky Gel for iPad/iPad mini Samsung Galaxy Size 6"- 10.5" iPad Car Holder

OHLPRO Tablet Holder for Car, iPad Holder for Car Air Vent Mount, Universal Dashboard Windshield 2-in-1 Cradle TPU Suction Sticky Gel for iPad/iPad mini Samsung Galaxy Size 6"- 10.5" iPad Car Holder

(1,961)
$22.99$15.09
FREE delivery Tue, Jun 17 on your first order
50+ bought in past month
Loading...
Seasonic Focus V4 GX-1000 (ATX3) - 1000W - 80+ Gold - ATX 3.0 & PCIe 5.1 Ready -Full-Modular -ATX Form Factor -Premium Japanese Capacitor -10 Year Warranty -Nvidia RTX 30/40 Super & AMD GPU Compatible

Seasonic Focus V4 GX-1000 (ATX3) - 1000W - 80+ Gold - ATX 3.0 & PCIe 5.1 Ready -Full-Modular -ATX Form Factor -Premium Japanese Capacitor -10 Year Warranty -Nvidia RTX 30/40 Super & AMD GPU Compatible

(22)
$423.35
FREE delivery Oct 8 - 10
Loading...
PNY NVIDIA Quadro RTX 4000 - The World’S First Ray Tracing GPU

PNY NVIDIA Quadro RTX 4000 - The World’S First Ray Tracing GPU

(203)
$670.57$399.99
FREE delivery Feb 23 - 27
Loading...
Microcontroller Board for Pico RP2040, Dual Core ARM Cortex M0+Processor Flexible Microcontroller Module (Transparent White)

Microcontroller Board for Pico RP2040, Dual Core ARM Cortex M0+Processor Flexible Microcontroller Module (Transparent White)

(0)
$9.82
FREE delivery Thu, Jun 19 on your first order
Loading...
Microcontroller Board for Pico RP2040, Dual Core ARM Cortex M0+Processor Flexible Microcontroller Module (Black)

Microcontroller Board for Pico RP2040, Dual Core ARM Cortex M0+Processor Flexible Microcontroller Module (Black)

(4)
$25.38
FREE delivery Thu, Jun 19 on your first order