Train a Custom YOLOv5 Model
After generating a version of your labeled object detection dataset, we are ready to train a model.
Here is a tutorial on training a YOLOv5 Model with the help of Roboflow's python package for easy data management.
Once you copy your dataset export code, continue on with the video tutorial to train your model:
Don't forget to save your model weights!
How to Save and Load Your Model Weights in Google Colab
Next Steps:
- Run YOLOv5 on Your Webcam
- Collect data from bad or imperfect model predictions, and improve your model with Active Learning: Implementing Active Learning