Model Training: YOLOv5-OBB

How to train a custom YOLOv5-OBB model with your Roboflow project.

Written by Mohamed Traore

Last published at: May 12th, 2022

Train a Custom YOLOv5-OBB Model

What is YOLOv5-OBB?

YOLOv5 for Oriented Object Detection (Roboflow Blog)

Oriented bounding boxes (OBB) are bounding boxes rotated to better fit the objects represented on an angle. Take a pill detection dataset for example. Using YOLOv5-OBB (or YOLOv5-obb) we are able to detect objects that are rotated on a given frame or image more tightly and accurately, preventing the capture of multiple objects or other objects in one bounding box.

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-OBB Model with the help of Roboflow's python package for easy data management.

There are 4 Ways to Export Your Dataset on Roboflow:

(1) Download to a Zip Folder

(2) Download with our python package - this is the most common method, as it allows you to programmatically download your images and use them to train a model with our model zoo.

(3) Download with a curl command

(4) Download by Raw URL

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