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!