YOLOv5: How to Train a Object Detection Model on Custom Dataset

Поделиться
HTML-код
  • Опубликовано: 20 сен 2024

Комментарии • 84

  • @NicolaiAI
    @NicolaiAI  Год назад +1

    Join My AI Career Program
    www.nicolai-nielsen.com/aicareer
    Enroll in My School and Technical Courses
    www.nicos-school.com

    • @hawkingT
      @hawkingT 25 дней назад

      can i get every command for google colab please ...any txt or just add in description here or reply in comment

  • @MasterChi__
    @MasterChi__ 2 года назад +4

    Cleanest follow through video I've seen on RUclips. Really excited for the second part!!!!

    • @NicolaiAI
      @NicolaiAI  2 года назад +1

      Thank you so much! Really appreciate it

    • @yuxrazafar8532
      @yuxrazafar8532 Месяц назад

      ​@@NicolaiAI can you please share the link to the colab notebook? I couldn't find it here in the description, and not on your github either.

  • @proomster
    @proomster 2 года назад +3

    Thank you so much for this video. 12 hours ago I didn't know where to start with making a YOLO model. I have just made my first model using your tutorial that labels vehicles and pedestrians from dashcam data. You have helped me a lot!

    • @NicolaiAI
      @NicolaiAI  2 года назад +1

      Thanks a lot for watching! Glad that i could help u

  • @tomlongson5903
    @tomlongson5903 2 года назад +1

    Thanks for posting all these amazing overviews, Nicolai!

    • @NicolaiAI
      @NicolaiAI  2 года назад

      Thanks a lot! Really appreciate it

  • @denizozdemir256
    @denizozdemir256 2 года назад +3

    where is the codes you use in the video,I couldnt find it in the github :/

  • @aryashinde6401
    @aryashinde6401 4 месяца назад +1

    Thank you very much for this video. It was very helpful for my project. Im extremely grateful to you😊

    • @NicolaiAI
      @NicolaiAI  4 месяца назад

      You are welcome 😊

  • @m4ts373
    @m4ts373 2 года назад +3

    Where did u get the colab Notebook from? I tried using the tutorial file from github and the Notbook from roboflow but they all were different. And great video ;) Its the first time i came in contact with computer vision and i really look forward to it

  • @shyampatel7038
    @shyampatel7038 Год назад +1

    Thank You for this video. very nice explanation !!

    • @NicolaiAI
      @NicolaiAI  Год назад +1

      Thanks for watching! If u are interested i have a yolov7 course aswell

    • @shyampatel7038
      @shyampatel7038 Год назад +1

      Yes i just saw that yolov7 course is there..
      Love form India❤

    • @NicolaiAI
      @NicolaiAI  Год назад +1

      @@shyampatel7038 Cool! Love from Denmark ❤❤

  • @patawee313
    @patawee313 2 года назад +2

    The walkthrough is very clear but I think your sound is a bit mumble. Looking forward to the next content.

    • @NicolaiAI
      @NicolaiAI  2 года назад

      Thanks a lot for watching! The audio has been improved a lot in the latest videos

  • @hectorserranocereijo2424
    @hectorserranocereijo2424 4 месяца назад

    Where can i have the full script? Thankyou!!

  • @AjibesinSamson
    @AjibesinSamson 2 года назад +2

    Good work. If you would like to evaluate the performance of the model developed using metrics such as mean average precision (mAP) and loss function, how do you go about that?

  • @jaredwaterman2689
    @jaredwaterman2689 2 года назад +1

    Amazing video mate

    • @NicolaiAI
      @NicolaiAI  2 года назад

      Thanks a lot for watching! Really appreciate it

  • @CapTainBPMT
    @CapTainBPMT Год назад +1

    So after I have followed this entire video once, if I wasn't happy with my confidence interval, should I go back to roboflow and label more images, then train the model again with the new data?

    • @NicolaiAI
      @NicolaiAI  Год назад +1

      Yeah that could be one way to do it. U can also analyse ur dataset to make sure its suitable. Trash in is trash out

  • @Team-fh2rr
    @Team-fh2rr Год назад +1

    Good day!
    We have some issues with object detection on custom datasets and would be appreciated any advice. Is it possible to have a 1-hours consultation?

    • @NicolaiAI
      @NicolaiAI  Год назад

      Yeah u Can reach out to me at thecodinglibraryyt@gmail.com

  • @petrsvoboda6249
    @petrsvoboda6249 2 года назад +1

    Hi, really nice tutorial! I would like to ask, if there is any way, to actually use this trained model with opencv?

  • @emmy_jon
    @emmy_jon 10 месяцев назад +1

    this is a good video. Is there a second video where we can deploy the .pt to openCV for realtime object detection

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

      Thanks a lot man! Yeah there should be some videos about it, or else I have my courses on my website where we go from start to end with everything in object detection and also tracking with yolov7 and yolov8

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

      @@NicolaiAI how do I get your courses?

    • @NicolaiAI
      @NicolaiAI  10 месяцев назад +1

      They are here in the pinned comments and video description!

    • @emmy_jon
      @emmy_jon 10 месяцев назад +1

      @@NicolaiAI Thank you🙏 can I get the yolov5 course, what I can see there is yolov7 & 8

    • @NicolaiAI
      @NicolaiAI  10 месяцев назад +1

      Those courses are the exact same content just with a newer model. Actually yolov5 is from the same creators as yolov8 so using the exact same framework. You can swap out yolov8 with yolov5 in a single line
      @@emmy_jon

  • @shashidharpatil4899
    @shashidharpatil4899 2 года назад +2

    Hey Buddy, thanks for the tutorial. I have one question ... After running detect.py for new images..it was able detect labels and show it as mug_0.7 ...But can you please tel how to store that mug_0.7 into an output file

    • @NicolaiAI
      @NicolaiAI  2 года назад

      Hi, thanks for watching! U can just take the bounding boxes which is the output from the model and store it in a CSV file as an example

  • @merondesta3895
    @merondesta3895 2 года назад

    thankyou sir that was useful. i trained using my own dataset but no matter how long i trained it the m@p will not get improved from 0.34 something like that i have used same data for yolov4 but it performs well

  • @kemalkartal366
    @kemalkartal366 9 месяцев назад +1

    hello,
    Can we add the object trained with the custom dataset to the other 80 object YOLO weights? As a single weight of 80+1. Can we increase the weight of the existing 80 objects?
    thanks.
    normally yolo weight consists of 80 objects.
    Can we add new objects to objects of this weight by training with custom datasets?

    • @NicolaiAI
      @NicolaiAI  9 месяцев назад

      Nope you can’t do that for the pretrained weights

    • @kemalkartal366
      @kemalkartal366 9 месяцев назад +1

      @@NicolaiAI thank you

  • @suhaipallaf6643
    @suhaipallaf6643 Год назад

    thank for this video...can you tell me how to get the location of bounding box pleas

  • @lsiddharthsingh
    @lsiddharthsingh 2 года назад

    Thank you so much brother. Your video really helped. Btw when is the 2nd part coming deploying the saved model in our own python code

  • @LordWildbeast
    @LordWildbeast Год назад

    where is the codes/colab notebook you use in the video?

  • @hassinijalil5533
    @hassinijalil5533 Год назад

    Hello sir
    Good work , i have a question the labels of yolov5 is it the same in yolo v4 and yolov3 italk about annotation ( the file.txt)
    i hope to answer me
    Thanks.

  • @赛尼木
    @赛尼木 2 года назад +1

    Greate work. I have one question. Is it possible to use your trained model in this video in a C++ program to perform object detection?

    • @NicolaiAI
      @NicolaiAI  2 года назад

      Thanks a Lot! Yeah u can use it in c++ too. I'll make a video in the future about that

  • @goodtechdoor
    @goodtechdoor 2 года назад

    I believe yolov5 does augmentation internally while training.

  • @theIndiangirlinEngland
    @theIndiangirlinEngland 2 года назад

    Hi, thanks for the tutorial. Can you please explain how to detect colour along with custom objects using yolo?

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

    where is the code and dataset for this?

  • @kunalpandya8468
    @kunalpandya8468 Год назад

    great video. can you provide code link as well?

  • @phonprasert
    @phonprasert 2 года назад

    Hi, Can you descripting for me how to labeling objects by drone image for deep learning.

  • @prakathisan5848
    @prakathisan5848 2 года назад +1

    how can we test it like we upload a picture and how it detects it?

    • @NicolaiAI
      @NicolaiAI  2 года назад

      We are going to create a video about how to deploy these models

  • @pedroivo6311
    @pedroivo6311 Год назад

    Ei cara, tudo bem? Onde eu altero os hyperparâmetros?

  • @tejalgoyal8940
    @tejalgoyal8940 Год назад

    where can i get this colab notebook?
    the one from the github is edited now

  • @agofpos
    @agofpos 2 года назад +1

    Can I know the specs of your computer? I encountered out of memory and I'm planning to upgrade my hardware

    • @NicolaiAI
      @NicolaiAI  2 года назад +1

      I7, GTX 1060 and 16 gb ram

  • @cyberhard
    @cyberhard 2 года назад +1

    Good tutorial. Can your model detect mugs of different colors?

    • @NicolaiAI
      @NicolaiAI  2 года назад

      We should try that! Most likely not or with low confidence score since we have only trained the model on the same mug

    • @cyberhard
      @cyberhard 2 года назад +1

      @@NicolaiAI That is why I asked. It would be an interesting experiment. I think it would speak volumes for YOLOV5 if it was still able to generalize.

    • @NicolaiAI
      @NicolaiAI  2 года назад

      I'll definitely try that! Thanks for the comment

    • @cyberhard
      @cyberhard 2 года назад

      @@NicolaiAI you're welcome.

  • @MiguelRodriguez-wn8mc
    @MiguelRodriguez-wn8mc Год назад

    good explanation see what I saw in the video I can train a model with roboflow and then use with ML5 and p5.js with cocossd or yolo

  • @lairajanereyes9936
    @lairajanereyes9936 2 года назад

    Do I really have to use roboflow huhu?

  • @joshuahearne3964
    @joshuahearne3964 2 года назад +1

    Roboflow is prompting me to pay to export my dataset?

    • @NicolaiAI
      @NicolaiAI  2 года назад +1

      Did u choose the public option?

    • @joshuahearne3964
      @joshuahearne3964 2 года назад +1

      @@NicolaiAI yeah realised that was my mistake, thanks

    • @NicolaiAI
      @NicolaiAI  2 года назад

      @@joshuahearne3964 great

  • @mohammadsari3003
    @mohammadsari3003 Год назад

    where can i access this code from ?

  • @jeanzyx1707
    @jeanzyx1707 2 года назад +1

    perfect

    • @NicolaiAI
      @NicolaiAI  2 года назад +1

      thanks a lot for watching! Really appreciate it

    • @jeanzyx1707
      @jeanzyx1707 2 года назад +1

      @@NicolaiAI that's my pleasure

  • @khalilhm3462
    @khalilhm3462 2 года назад

    hi dude please i wanna use stero camera for detection object and thier location please help

    • @abderrahmenmelliti9627
      @abderrahmenmelliti9627 2 года назад +1

      Hi Khalil did you finish this task ? I have the same task that I need to do, can i ask you some questions please?

    • @khalilhm3462
      @khalilhm3462 2 года назад

      @@abderrahmenmelliti9627 yes i did

    • @broke3039
      @broke3039 2 года назад

      @@khalilhm3462 do you have your project on github

  • @thelaughtermedia9702
    @thelaughtermedia9702 2 года назад

    code for potholes pls

  • @karthikeyan3016
    @karthikeyan3016 Год назад

    👍

  • @khalilhm3462
    @khalilhm3462 2 года назад +1

    please if you can see my message in the emails just let me know thanks (im member in the Channel )

    • @NicolaiAI
      @NicolaiAI  2 года назад +1

      Hi i have just seen the mail, I'll get to it :)

  • @oumaimabenabdelmalek3096
    @oumaimabenabdelmalek3096 Год назад

    where is the codes you use in the video,I couldnt find it in the github :/