From 592f421663d95bd5f3ad0b75520471225eb7edc0 Mon Sep 17 00:00:00 2001 From: Francesco Mattioli Date: Mon, 12 Aug 2024 22:51:06 +0200 Subject: [PATCH] Fixed multiscale preprocess_batch (#15392) --- ultralytics/models/yolo/detect/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/models/yolo/detect/train.py b/ultralytics/models/yolo/detect/train.py index 65b7efff..27d172b2 100644 --- a/ultralytics/models/yolo/detect/train.py +++ b/ultralytics/models/yolo/detect/train.py @@ -60,7 +60,7 @@ class DetectionTrainer(BaseTrainer): if self.args.multi_scale: imgs = batch["img"] sz = ( - random.randrange(self.args.imgsz * 0.5, self.args.imgsz * 1.5 + self.stride) + random.randrange(int(self.args.imgsz * 0.5), int(self.args.imgsz * 1.5 + self.stride)) // self.stride * self.stride ) # size