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

single_channel
Clea Parcerisas 1 year ago
parent d221ace5a3
commit ba4c33dc53

@ -157,7 +157,7 @@ class BaseValidator:
self.dataloader = self.dataloader or self.get_dataloader(self.data.get(self.args.split), self.args.batch) self.dataloader = self.dataloader or self.get_dataloader(self.data.get(self.args.split), self.args.batch)
model.eval() 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() dt = Profile(), Profile(), Profile(), Profile()
n_batches = len(self.dataloader) n_batches = len(self.dataloader)

Loading…
Cancel
Save