This repository contains the code for Custom Object Detection Training using YOLOv5 blogpost.
The Custom_Object_Detection_using_YOLOv5.ipynb Jupyter Notebook contains all the code and is one-click runnable. The experiments available in the notebook are:
- Training a small YOLOv5 model.
- Training a medium YOLOv5 model.
- Training a medium YOLOv5 model by freezing the first 15 blocks.
You can run the notebook on Google Colab as well.
Want to become an expert in AI? AI Courses by OpenCV is a great place to start.