From ba4c33dc531d32073f0ef4a4fbba45d8892bcfa2 Mon Sep 17 00:00:00 2001 From: Clea Parcerisas Date: Thu, 24 Aug 2023 10:54:18 +0200 Subject: [PATCH] changed the imgsize on enginge/validator to meet the grayscale for prediction --- ultralytics/engine/validator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/validator.py b/ultralytics/engine/validator.py index ff50b52..a4bc790 100644 --- a/ultralytics/engine/validator.py +++ b/ultralytics/engine/validator.py @@ -157,7 +157,7 @@ class BaseValidator: self.dataloader = self.dataloader or self.get_dataloader(self.data.get(self.args.split), self.args.batch) model.eval() - model.warmup(imgsz=(1 if pt else self.args.batch, 3, imgsz, imgsz)) # warmup + model.warmup(imgsz=(1 if pt else self.args.batch, 1, imgsz, imgsz)) # warmup dt = Profile(), Profile(), Profile(), Profile() n_batches = len(self.dataloader)