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

is_docker() fix for show=True predict bug (#218)

Browse Source 
Co-authored-by: Hardik Dava <39372750+hardikdava@users.noreply.github.com>
Co-authored-by: Onuralp Sezer <thunderbirdtr@fedoraproject.org>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-01-11 00:37:07 +01:00
committed by GitHub
parent 7f9d7142c2
commit 88c9418087
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/yolo/engine/predictor.py
View File

@ -124,9 +124,8 @@ class BasePredictor:
# Dataloader
bs = 1 # batch_size
if self.args.show:
self.args.show = check_imshow(warn=True)
if webcam:
self.args.show = check_imshow(warn=True)
self.dataset = LoadStreams(source,
imgsz=imgsz,
stride=stride,