changed the imgsize on enginge/validator to meet the grayscale for prediction

This commit is contained in:
Clea Parcerisas
2023-08-24 10:54:18 +02:00
parent d221ace5a3
commit ba4c33dc53

View File

@ -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)