HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.0.175 StreamLoader wait for missing frames (#4814)

Browse source
This commit is contained in:
Glenn Jocher 2023-09-10 23:59:43 +02:00 • committed by GitHub
parent dd0782bd8d
commit 3c88bebc95
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 197 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/predictor.py
View file

@ -207,7 +207,7 @@ class BasePredictor:
self.dataset = load_inference_source(source=source,
imgsz=self.imgsz,
vid_stride=self.args.vid_stride,
stream_buffer=self.args.stream_buffer)
buffer=self.args.stream_buffer)
self.source_type = self.dataset.source_type
if not getattr(self, 'stream', True) and (self.dataset.mode == 'stream' or # streams
len(self.dataset) > 1000 or # images