How to Use MMDetection | Train RTMDet on a Custom Dataset

Подписаться 37 тыс.

Просмотров 15 тыс.

50% 1

Dive into the world of computer vision with this comprehensive tutorial on training the RTMDet model using the renowned MMDetection library. Whether you're just starting out or looking to refine your skills, this guide offers a deep dive into the OpenMMLab ecosystem, hands-on installation steps, and practical insights into training on custom datasets.
Chapters:
- 00:00 Introduction
- 00:29 What is MMDetection and RTMDet
- 01:24 OpenMMLab Libraries Installation
- 04:54 Inference with Pre-trained COCO Model
- 08:14 Downloading a Dataset from Roboflow Universe
- 10:02 Preparing Custom MMDetection Configuration File
- 11:30 Train RTMDet and Analyze the Metrics
- 13:08 Evaluating the RTMDet Model with Supervision
- 14:24 Conclusions
Resources:
- 🌏 Roboflow: roboflow.com
- 📚 Roboflow Notebooks Repository: github.com/roboflow/notebooks
- 🌌 Roboflow Universe: universe.roboflow.com
- 📓 RTMDet Object Detection Google Colab: colab.research.google.com/git...
- 🗞️ RTMDet: An Empirical Study of Designing Real-Time Object Detectors Arxiv paper: arxiv.org/pdf/2212.07784.pdf
- 🗞️How to Train RTMDet on a Custom Dataset blog post: blog.roboflow.com/how-to-trai...
- 💻 MMDetection GitHub repository: github.com/open-mmlab/mmdetec...
- 🖼️ Double Twelve Dominoes object detection dataset: universe.roboflow.com/pip-tra...
Remember to like, comment, and subscribe for more content on AI, computer vision, and the latest technological breakthroughs! 🚀
Stay updated with the projects I'm working on at github.com/roboflow and github.com/SkalskiP! ⭐

Наука

Опубликовано:

5 авг 2024

Ссылка:

Скачать:

Готовим ссылку...

Добавить в:

Мой плейлист

Посмотреть позже

Комментарии : 57

@suphotnarapong355 11 месяцев назад

Thank you so much for sharing the knowledge I've been searching for. Your willingness to share this information has been incredibly helpful to me, and I truly appreciate it. It's made a significant difference in my understanding and learning. Thanks again!

@JemiloII Месяц назад

The fact that you care about licensing that helps us all out, instant subscribe. Pretty tired of seeing AGPL Licensed code/models being used over and over.

@hartmannr76 11 месяцев назад

Hey, cool! You're using my domino dataset :) Thrilled to see it being used in something like this!!

@Roboflow 11 месяцев назад

So awesome! Shame I couldn’t contact you before. I’d love to test the model on the video input.

@Lena-qk7qq 5 месяцев назад

Awesome video, thanks!

@user-yl3fl8bs1b 11 месяцев назад

You're welcome for the helpful advice. I will try using it in my project.

@Roboflow 11 месяцев назад

Awesome! Please, let me know if you like it. And btw, was the LICENSE important factor for you?

@thanachai.j 11 месяцев назад

Thank for teaching .❤

@Roboflow 11 месяцев назад

My pleasure!

@top-5-travel 8 месяцев назад

I have tried the google colab and it is not working. It doesn't install mmcv. Any ideas how to correct this?

@illyaaverchenko2263 3 месяца назад

@hyperion6821 1 месяц назад I solved by installing torch==2.0.1 and some other stuff before installing all the mmdetection stuff. !pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 torchdata==0.6.1 torchtext==0.15.2 fastapi kaleido python-multipart uvicorn lida torchdata

@manujkumarjoshi9342 10 месяцев назад

Beautiful

@Roboflow 10 месяцев назад

Thank you!

@user-jj8qz2vy7o 8 месяцев назад

I've been trying to get mmdet working in a google colab project, and I keep running into an error where it fails to build a wheel for mmcv. The project from the video appears to have been running fine a few months back, but when I am running the code now, it gets the error I've been getting. Is anyone getting the same error? Does anyone have any insight or fixes?

@seanolivieri4829 8 месяцев назад

Me too: ERROR: Failed building wheel for mmcv ERROR: Could not build wheels for mmcv, which is required to install pyproject.toml-based projects ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. yfinance 0.2.31 requires requests>=2.31, but you have requests 2.28.2 which is incompatible. ModuleNotFoundError: No module named 'mmdet'

@riley8705 8 месяцев назад

Has anyone found a solution to this

@illyaaverchenko2263 3 месяца назад

I solved by installing torch==2.0.1 and some other stuff before installing all the mmdetection stuff. !pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 torchdata==0.6.1 torchtext==0.15.2 fastapi kaleido python-multipart uvicorn lida torchdata

@zeeshanrizvi3350 2 месяца назад

Interestingly, I just solved this error a while ago. Try changing the version of torch to 2.0.0 and then run the command to install mmcv again. Hope it helps!

@murphy1162 11 месяцев назад

Great job!!! I just need it

@Roboflow 11 месяцев назад

We love to hear that! 🔥

@SkalskiP 11 месяцев назад

Thank you! 🙏

@tektronix475 11 месяцев назад

Does it have image anomaly detection?

@Roboflow 11 месяцев назад

@@tektronix475 when you say anomaly, do you mean fault detection?

@murphy1162 11 месяцев назад

@@Roboflowactually i am waiting for the latest SOTA swimTransformer for objectdetection!

@valliebear5706 2 месяца назад

Hi good tutorial, can I ask about the train loss and validation loss you showed the loss graph so which is the validation and train loss? is it possible to open the json and manually fetch it and use matplot? because if I use the mmdetection 3.* it does not prove the validation loss. thank you

@abenezeryonathan3971 8 месяцев назад

Thank you so much for this tutorial. I wanted to implement tracking but I am having problem with MMTracker and MMYOLO. Can you do MMDetection and MMTracker model training with custom dataset in the future.... Please 🙏

@AdarshSingh-rm6er 3 месяца назад

Hey thanks for the informative video. It was really intuitive and helped me to understand more about mmdetection model. I was using mmdetection model for face detection and getting error AttributeError: 'list' object has no attribute 'pred_instances' when i am training the model on the below line : detections = sv.Detections.from_mmdetection(results) Your help will be greatly appriciated.

@none-hr6zh 8 месяцев назад

how to make mmd datasets if I have already datasets in yolo and coco format.

@danielarmandovidalsoroa7787 3 месяца назад

Hello, can we also deploy the custom-trained RTMdet model in roboflow? Or it does not work? Could you provide the code?

@amarnathreddysurapureddy9111 3 месяца назад

Is it possible to train the Annotated Custom dataset using pytorch? I want train a model using pytorch but annotated images from roboflow/cvit/supervision. I have 600 images. I Annotated 600 iamges with different labels .I want classify dog, cat, rabbit, cow and mouse. Is it possible?

@tryingtobeproductive 11 месяцев назад

7:35 I chuckled haha

@Roboflow 10 месяцев назад

Me too! Haha

@fredcdf75 10 месяцев назад

Thank you for this great tutorial! About the licenses, I think rtmdet is under Apache license only in mmdet but not in mmyolo (which is supposed to be a faster training version). In mmyolo rtmdet is under GPL v3 license. Please tell me if I am wrong I'd rather use mmyolo :)

@Roboflow 10 месяцев назад

This is really good question. Up until now thought that MMYolo offers only after training of MMDet model. So same model as MMDet. And only fast training code is under GPL.

@fredcdf75 10 месяцев назад

@@Roboflow Yes you're right the model should be the same but different training code. Anyway thanks again for taking the time to make these tutorials and replying to comments. I'm still a beginner and it helps a lot 👍

@gauravlochab9614 11 месяцев назад

very useful thanks! Can you make a tutorial on mm library as in whole

@Roboflow 11 месяцев назад

Hm... There are 30+ libraries in that ecosystem. That would probably be a very long video. What exactly would you like to learn?

@gauravlochab9614 11 месяцев назад

@@Roboflow mmsegmentation and mmdetection context -Given a detection problem and the need to select an appropriate model based on data health check in Roboflow, how should I proceed? For instance, when dealing with factors such as small boxes or high resolution within the data, how do I determine the suitable model in mmdetection and its necessary tuning? Your guidance is greatly appreciated. I'm an avid consumer of your content and a daily user of Roboflow for data preparation. Thank you.

@hysskm 10 месяцев назад

@Roboflow I want to know how to convert Our dataset's annotation format into coco format

@marcelopedras88 6 месяцев назад

It no longer works on Colab. Can you solve this? ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. ipython 7.34.0 requires jedi>=0.16, which is not installed. lida 0.0.10 requires fastapi, which is not installed. lida 0.0.10 requires kaleido, which is not installed. lida 0.0.10 requires python-multipart, which is not installed. lida 0.0.10 requires uvicorn, which is not installed. cvxpy 1.3.3 requires setuptools>65.5.1, but you have setuptools 60.2.0 which is incompatible. google-colab 1.0.0 requires requests==2.31.0, but you have requests 2.28.2 which is incompatible. yfinance 0.2.36 requires requests>=2.31, but you have requests 2.28.2 which is incompatible.

@Roboflow 6 месяцев назад

Please create issue here and describe your problems: github.com/roboflow/notebooks/issues

@Asadkhan-nz6qm 10 месяцев назад

Hello bro i have a question to the best of your knowledge which model is best for medical images?

@Roboflow 10 месяцев назад

Are we talking about segmentation or detection?

@diemagic5140 5 месяцев назад

Google Colab is not working please update it

@illyaaverchenko2263 3 месяца назад

@lawrencerodriques7153 10 месяцев назад

Hey can you create a tutorial for smart Surveillance which detect human intrusion classify between authorised and unauthorised intruders and notify security person usign YOLO V8 please

@Roboflow 10 месяцев назад

How do I know which intrusion is authorized and unauthorized?

@lawrencerodriques7153 10 месяцев назад

@@Roboflow You will have to create authorised persons dataset and then verify like attendance system

@lawrencerodriques7153 10 месяцев назад

@@Roboflow please create it 🙏🙏

@seanolivieri4829 5 месяцев назад

ERROR: Failed building wheel for mmcv