From 4cef9d047e66c3795f79af0d78eaf3c492eb0925 Mon Sep 17 00:00:00 2001 From: dagokl Date: Wed, 11 Dec 2024 21:05:27 +0100 Subject: [PATCH] Fix validation callbacks not triggered during OBB model training (#18175) Co-authored-by: Glenn Jocher --- ultralytics/models/yolo/obb/train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 + )