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

Clean up unused imgsz (#7771)

Browse source
This commit is contained in:
Laughing 2024-01-24 01:50:01 +08:00 • committed by GitHub
parent f56dd0f48e
commit 67ae86f006
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 10 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/predictor.py
View file

@ -226,7 +226,7 @@ class BasePredictor:
else None
)
self.dataset = load_inference_source(
source=source, imgsz=self.imgsz, vid_stride=self.args.vid_stride, buffer=self.args.stream_buffer
source=source, vid_stride=self.args.vid_stride, buffer=self.args.stream_buffer
)
self.source_type = self.dataset.source_type
if not getattr(self, "stream", True) and (