xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Clea Parcerisas
2023-08-24 10:54:18 +02:00
parent d221ace5a3
commit ba4c33dc53
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/engine/validator.py
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) 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)