How to Use MMDetection | Train RTMDet on a Custom Dataset

แชร์
ฝัง
  • เผยแพร่เมื่อ 7 ก.ค. 2024
  • 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! ⭐
  • วิทยาศาสตร์และเทคโนโลยี

ความคิดเห็น • 57

  • @JemiloII
    @JemiloII 23 วันที่ผ่านมา

    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.

  • @suphotnarapong355
    @suphotnarapong355 10 หลายเดือนก่อน +1

    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!

  • @hartmannr76
    @hartmannr76 10 หลายเดือนก่อน +2

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

    • @Roboflow
      @Roboflow  10 หลายเดือนก่อน +1

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

  • @Lena-qk7qq
    @Lena-qk7qq 4 หลายเดือนก่อน

    Awesome video, thanks!

  • @thanachai.j
    @thanachai.j 10 หลายเดือนก่อน +1

    Thank for teaching .❤

    • @Roboflow
      @Roboflow  10 หลายเดือนก่อน

      My pleasure!

  • @user-yl3fl8bs1b
    @user-yl3fl8bs1b 10 หลายเดือนก่อน +2

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

    • @Roboflow
      @Roboflow  10 หลายเดือนก่อน +1

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

  • @manujkumarjoshi9342
    @manujkumarjoshi9342 9 หลายเดือนก่อน

    Beautiful

    • @Roboflow
      @Roboflow  9 หลายเดือนก่อน

      Thank you!

  • @murphy1162
    @murphy1162 10 หลายเดือนก่อน +2

    Great job!!! I just need it

    • @Roboflow
      @Roboflow  10 หลายเดือนก่อน

      We love to hear that! 🔥

    • @SkalskiP
      @SkalskiP 10 หลายเดือนก่อน

      Thank you! 🙏

    • @tektronix475
      @tektronix475 10 หลายเดือนก่อน

      Does it have image anomaly detection?

    • @Roboflow
      @Roboflow  10 หลายเดือนก่อน

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

    • @murphy1162
      @murphy1162 10 หลายเดือนก่อน

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

  • @user-jj8qz2vy7o
    @user-jj8qz2vy7o 7 หลายเดือนก่อน +7

    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
      @seanolivieri4829 7 หลายเดือนก่อน +1

      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
      @riley8705 7 หลายเดือนก่อน

      Has anyone found a solution to this

    • @illyaaverchenko2263
      @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
      @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!

  • @abenezeryonathan3971
    @abenezeryonathan3971 7 หลายเดือนก่อน

    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 🙏

  • @top-5-travel
    @top-5-travel 7 หลายเดือนก่อน +3

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

    • @illyaaverchenko2263
      @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

  • @valliebear5706
    @valliebear5706 หลายเดือนก่อน

    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

  • @AdarshSingh-rm6er
    @AdarshSingh-rm6er 2 หลายเดือนก่อน

    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
    @none-hr6zh 7 หลายเดือนก่อน

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

  • @tryingtobeproductive
    @tryingtobeproductive 10 หลายเดือนก่อน +2

    7:35 I chuckled haha

    • @Roboflow
      @Roboflow  9 หลายเดือนก่อน

      Me too! Haha

  • @danielarmandovidalsoroa7787
    @danielarmandovidalsoroa7787 2 หลายเดือนก่อน

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

  • @fredcdf75
    @fredcdf75 9 หลายเดือนก่อน +1

    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
      @Roboflow  9 หลายเดือนก่อน +1

      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
      @fredcdf75 9 หลายเดือนก่อน

      @@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
    @gauravlochab9614 10 หลายเดือนก่อน +1

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

    • @Roboflow
      @Roboflow  10 หลายเดือนก่อน

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

    • @gauravlochab9614
      @gauravlochab9614 10 หลายเดือนก่อน

      ​@@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
      @hysskm 9 หลายเดือนก่อน

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

  • @amarnathreddysurapureddy9111
    @amarnathreddysurapureddy9111 2 หลายเดือนก่อน

    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?

  • @Asadkhan-nz6qm
    @Asadkhan-nz6qm 9 หลายเดือนก่อน +1

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

    • @Roboflow
      @Roboflow  9 หลายเดือนก่อน

      Are we talking about segmentation or detection?

  • @lawrencerodriques7153
    @lawrencerodriques7153 9 หลายเดือนก่อน

    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
      @Roboflow  9 หลายเดือนก่อน

      How do I know which intrusion is authorized and unauthorized?

    • @lawrencerodriques7153
      @lawrencerodriques7153 9 หลายเดือนก่อน

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

    • @lawrencerodriques7153
      @lawrencerodriques7153 9 หลายเดือนก่อน

      @@Roboflow please create it 🙏🙏

  • @marcelopedras88
    @marcelopedras88 5 หลายเดือนก่อน

    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
      @Roboflow  5 หลายเดือนก่อน +1

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

  • @diemagic5140
    @diemagic5140 4 หลายเดือนก่อน +2

    Google Colab is not working please update it

    • @illyaaverchenko2263
      @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

  • @seanolivieri4829
    @seanolivieri4829 4 หลายเดือนก่อน +1

    ERROR: Failed building wheel for mmcv

    • @hyperion6821
      @hyperion6821 4 หลายเดือนก่อน +2

      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 yfinance
      still gets a conflict for the requests package but I didn't get any trouble with that.

    • @diemagic5140
      @diemagic5140 4 หลายเดือนก่อน

      @@hyperion6821 It works thanks

    • @danielarmandovidalsoroa7787
      @danielarmandovidalsoroa7787 2 หลายเดือนก่อน

      @@hyperion6821 It worked for me thank you 🙏

    • @michaelburford2151
      @michaelburford2151 หลายเดือนก่อน

      @@hyperion6821 worked !

    • @truongnguyennhat9356
      @truongnguyennhat9356 9 วันที่ผ่านมา

      @@danielarmandovidalsoroa7787 can u shere your script

  • @me111ok
    @me111ok 4 หลายเดือนก่อน +1

    Amazing how quickly tutorial on ML get outdated. But thanks anyways

    • @Roboflow
      @Roboflow  4 หลายเดือนก่อน

      At this point it is a full time job to be up to date ;)