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

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
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