Reference for ultralytics/models/yolo/obb/train.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/train.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.obb.train.OBBTrainer
Bases: DetectionTrainer
A class extending the DetectionTrainer class for training based on an Oriented Bounding Box (OBB) model.
Example
Source code in ultralytics/models/yolo/obb/train.py
get_model
Return OBBModel initialized with specified config and weights.
Source code in ultralytics/models/yolo/obb/train.py
get_validator
Return an instance of OBBValidator for validation of YOLO model.
Source code in ultralytics/models/yolo/obb/train.py
📅 Created 10 months ago ✏️ Updated 1 month ago