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

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

Browse Source
This commit is contained in:
Clea Parcerisas
2023-08-24 11:09:35 +02:00
parent ba4c33dc53
commit 95e2c2424a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/engine/predictor.py
View File

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