/YoloV5-Custom-Object-Detection

Custom Object Detection With YoloV5

Primary LanguageJupyter Notebook

Custom Object Detection With YoloV5

This repository contains a notebook to train your own custom object detector using YOLOv5. I have used the code of Ultralytics to train the model. Make sure to check their repository also. It's great.

I have posted a detailed tutorial in medium. Here's the link:

Training:

Tensorboard results :

If you have any questions, recommendations or critiques, I can be reached via Twitter or via my mail. Feel free to reach out to me.