diff --git a/ultralytics/models/yolo/obb/train.py b/ultralytics/models/yolo/obb/train.py index 18def219..9c2b08b9 100644 --- a/ultralytics/models/yolo/obb/train.py +++ b/ultralytics/models/yolo/obb/train.py @@ -39,4 +39,6 @@ class OBBTrainer(yolo.detect.DetectionTrainer): def get_validator(self): """Return an instance of OBBValidator for validation of YOLO model.""" self.loss_names = "box_loss", "cls_loss", "dfl_loss" - return yolo.obb.OBBValidator(self.test_loader, save_dir=self.save_dir, args=copy(self.args)) + return yolo.obb.OBBValidator( + self.test_loader, save_dir=self.save_dir, args=copy(self.args), _callbacks=self.callbacks + )