Batch inference fixes (#8854)

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

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