How to Train YOLOv5 on a Custom Dataset

YOLOv5 is the latest evolution in the YOLO family of object detection models. It's the first YOLO implementation native to PyTorch (rather than Darknet) and emphasizes ease of use and quickness of training and inference. This YOLOv5 tutorial shows you how to train the model on your own dataset in Python.

Tools:

YOLOv5 Details:

Labeling guides for creating bounding boxes:

Explore popular templates to kickstart your project:

[milesgiehtbrock]

Dude thanks for speaking in a way that teaches me instead of just assumes I already know this stuff

[raunakmitra]

Brilliantly done! You know how to keep the viewers glued to your video.

[philippehuet]

Great tutorial, maybe the best I have seen on Yolo. Thanks for sharing the Colab notebook

[kunlunwang]

Best YOLO tutorial I've seen on Youtube so far! Thank you!

[A_tavv]

this was the best video in the detection area by YOLO, believe me I tried a bunch of videos, non of them worked like this Thanks truly.

[luqmansen]

I'm about to follow a yolo3 tutorial, and then I found the yolo4 tutorial, and then I found yolo5 tutorial, omg Thanks roboflow.

[tylersnard]

This worked GREAT for me! The best part was that it showed my training images and their labels, and it made me realize my training set was mis-labeled :)

[dianavelychko]

this video is so transparent and really easy to understand and follow

[tarot-ooff]

I liked the shaving part and selfie part ..by the way ..this was so fun learning that hard skill in a funniest way..thankur so much sir..

[hctpzsw]

this guy is still a legend to this day

[russf]

Apart from the nits, noted below, THANKS for a great video, delivered in such an upbeat way!

[bbb-thvu]

i love your facial expressions. thanks. i dont know anything yet but hopefully i am planning to go into this, this summer.

[mrpoeky]

Thank you! for the clear explanation :)

[pedroh.fonseca]

Best tutorial ever!! Thankss Joseph!! ;D

[sorrybird]

Thanks for your awesome sharing. Helps us a lot!

[azerioauditore]

He is funny af. And obvio the tutorial is awesome, thats very universal of him.

[abdullateefagbaje]

This is a great video. I have a question, what if I want to use TensorFlow instead of PyTorch for YOLOv5 or 4?

[delibrain]

Hi Joseph .A great tutorial. The dataset you used was heavily unbalanced. The augmentation is performed on all images combined and not the ones that were misrepresented(WBC AND platelets).
Don't you thing that the trained model will be biased towards RBC?
Could you give some suggestions how to cope with unbalanced dataset?
or should I perform augmentation on the least represented classes?

[hackercop]

Sir you are amazing thanks, you make we want to get roboflow Pro! Keep it up!

[francescopaganelli]

Is it possible detect or recognize different person like different classes? Thanks great job!