Train a custom object detection model using your data

Learn how to train a custom object detection model for Raspberry Pi to detect less common objects like versions of a logo using your own collection of data.

00:00 Introduction
00:49 The 3 steps of training a custom model
01:24 Step 1: Create a training dataset
04:01 Step 2: Train a custom model with TensorFlow Lite Model Maker
09:03 Step 3: Deploy the custom model to Raspberry Pi
11:08 What’s next

#TensorFlow #MachineLearning #ML #RaspberryPi #EdgeAI

product: TensorFlow - TensorFlow Lite, TensorFlow - General; fullname: Khanh LeViet;

[silvanovich]

Thank you for sharing this straight forward approach to object detection. I have read book chapters and looked at a variety of web pages describing how this can be done, and while I am grateful to those who shared their knowledge, this video was at a level that I can understand. It explains the concepts without resorting to jargon and having to use a .tfrec input format.

[bandara-skrf]

while I am downloading "!pip install tflite-model-maker" my Disk space is not enough for it. how can i manage it ?

[vladimirputin]

Looks like this project stopped working, just getting a bunch of errors during installation

[margramiof]

This video series is pure gold. Thank you for sharing and explaining the power of this framework. Just a comment, the colab notebook in the above link has an error in the line which installs the required packages. Instead of :

!pip install -q tflite-model-maker-nightly
!pip install -q tflite-support-nightly

as shown in the video.

It says:
!pip install -q tflite-model-maker
!pip install -q tflite-support

In my case, those commands took a long time (almost timeout ) and they did not work.

[mbrraghu]

hi,
!pip install -q tflite-model-maker-nightly
!pip install -q tflite-support-nightly
is killing the colab space, more than 50 GB. What is the solution?

[jitesh_helloworld]

Suddenly, model-maker package is not getting installed in the colab notebook you shared. Note book was working perfectly fine till last week.

[aundo]

The colab notebook doesn't work anymore, !pip install tf-model-maker is stuck in an infinite download loop. Why is that?

[nabeelmalik.]

When i execute !pip install -q tflite-model-maker-nightly

It start downloading dependencies and never stop until disk low message has occured.

Any solution to this problem.?

[abhijitsutar]

I'm using the notebook currently in colab with python 3.9. While installing tflite-model-maker it's installing all the versions of tf-gpu >=2.6, which leads to out of storage situation in colab, without properly installing the dependencies. If you checked my comment then please reply back . Thank you

[denizcanturk]

Great explanation and most importantly whatever he said works properly... Great job and thanks..

[ungifted]

As of 31, Mar, 2024, tflite model maker can not be installed, due to compatibility issue with Python version 3.10 (According several report in Github and other sites).
Is there any solution ???

[josenoventaeum]

You're so objective and clear that everything sounds veeeery easy!

[TechCree]

I am really enthusiastic about your contribution and hope for some more videos of this kind. My goal is that the Pi can later output an acoustic message when it has detected an object. For example, if a person were to enter the room in which a blind person is located, I want the person to be recognized and I can use Python and possibly espeak to communicate that "Pete" has entered the room. Or at a danger point it would be possible to get a message when an object is approaching. Or, if necessary, switch lights on and off via GPIO interfaces or start an LED warning sequence. My head bursts when I think about the possibilities it offers. Please please continue in this direction !!!

[richarballesteros]

i have been 2 years trying to do this and finally works

[rjtffqe]

Can't use your project for training know . Could you please fix the error of installation ?

[Terrar]

can you please make a video of creating an absolutely empty model and training it from scratch (not transfer), even if thousands of pictures are required? It would be very useful for our scientific studies, thanks!

[KarthickRobo]

thanks a lot i am new to computer vison its easy to start

[shooby]

the packages won't install on google colab

[Auditor]

thank you a lot, you made understanding this a lot easier.

[thupilcharwaka]

ERROR: Could not install packages due to an OSError: [Errno 28] No space left on device