How to train an object detection model - ML on Raspberry Pi with MediaPipe

preview_player
Показать описание
Learn how to create your own object detection model that you can deploy to your Raspberry Pi device. Paul Ruiz, a Senior Developer Advocate, shares a high level approach to training a model so that you can prototype your ideas.

Resources:

#ML #MediaPipe
Рекомендации по теме
Комментарии
Автор

Hello Google Dev Team! This new way of training with Mediapipe is excellent, especially for deploying on Raspberry Pi 4. I have trained a custom network to identify industrial vehicles within a factory, and it works ✅. I would like to improve the processing speed because sometimes it is a bit slow. I would like to know if you can do a video explaining how to train a network for a tflite model to use Coral AI on Raspberry pi for improve a speed, and do the corresponding deploy 😁. I thank you very much, congrats! Excellent work.

santiagolassog
Автор

Thanks for the great video. The tips on how to fiddle with Label Studio's labels.json file output (make sure you have 0 be a background class, etc.) were extremely helpful. (FWIW, my current project is a Pi + a battery-powered water gun. It's meant to detect and deter squirrels trying to eat oranges on our orange tree.)

I think a great addition to the series would be some intro tips on how to adjust hyperparameters to potentially improve accuracy. Please consider putting it on the to-do list!

oliverexcellent
Автор

Hi Paul, thanks for your clear instructions. I was able to create a simple model, deployed to Pi and ran it successfully. I also managed to compile it to a Edgetpu model using Kang’s colab by down grade to tensorflow 2.13.0. It did compiled. Unfortunately, it failed to run on Pi for some runtime errors. Would you please provide instructions on how to compile it with mediapipe support. Thanks!

thomashu
Автор

Hello, your video is really useful, thank you. There is one point I don't understand.
I see that you created a label for the background. Are there background images in your dataset and have you labeled them? Or do we just need to create it as a label and automatically the unlabeled area in any image becomes the background?

hasankesoglu
Автор

at 12:39, there are some other commands "qat_hparams = that never got executed in this tutorial? What's that about?

Thanks for this awesome guide too! That was very helpful. I'm working on a calibration fixture, and ideally I would like to import this custom model into OpenCV.

frankbraker
Автор

I have trained a model now. Can I add new photos and categories for training based on this model? If so, will previously trained parts be retrained?

mlfbsts
Автор

i get error No module named 'keras.src.engine', how to fix this

muhammadnorirfan
Автор

is there any alternative for label studio? i don't have ubuntu devices

qbotx
Автор

when importing from mediapipe_model_maker import object_detector I get error ModuleNotFoundError: No module named 'keras.src.engine'

ukaszlebiecki