From 95e2c2424a8eb9eae49b59ee98a4c5fea1a66611 Mon Sep 17 00:00:00 2001 From: Clea Parcerisas Date: Thu, 24 Aug 2023 11:09:35 +0200 Subject: [PATCH] changed the imgsize on enginge/prediction to meet the grayscale for prediction --- ultralytics/engine/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/predictor.py b/ultralytics/engine/predictor.py index 8b10980..c42c1b5 100644 --- a/ultralytics/engine/predictor.py +++ b/ultralytics/engine/predictor.py @@ -236,7 +236,7 @@ class BasePredictor: # Warmup model if not self.done_warmup: - self.model.warmup(imgsz=(1 if self.model.pt or self.model.triton else self.dataset.bs, 3, *self.imgsz)) + self.model.warmup(imgsz=(1 if self.model.pt or self.model.triton else self.dataset.bs, 1, *self.imgsz)) self.done_warmup = True self.seen, self.windows, self.batch, profilers = 0, [], None, (ops.Profile(), ops.Profile(), ops.Profile())