How to Train TensorFlow Lite Object Detection Models Using Google Colab | SSD MobileNet
Edje Electronics
@edjeelectronicsAbout
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

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 and Nintendo Switch Lite - Sonic Peel Out

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

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

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)

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

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

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

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

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

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

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















