xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Batch inference fixes (#8854)

Browse Source
This commit is contained in:
Burhan 2024-03-11 12:14:03 -04:00 committed by GitHub
parent 5893d4483e
commit ff8ae0d2e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/engine/predictor.py
View File

@ -318,7 +318,7 @@ class BasePredictor:
else:
frame = getattr(self.dataset, "frame", 0) - len(self.results) + i
self.txt_path = self.save_dir / "labels" / (p.stem + f"_{frame}" if is_video[i] else "")
self.txt_path = self.save_dir / "labels" / (p.stem + (f"_{frame}" if is_video[i] else ""))
string += "%gx%g " % im.shape[2:]
result = self.results[i]
result.save_dir = self.save_dir.__str__() # used in other locations