From ff8ae0d2e25528badc93277c4c904f5d09f9eec4 Mon Sep 17 00:00:00 2001 From: Burhan <62214284+Burhan-Q@users.noreply.github.com> Date: Mon, 11 Mar 2024 12:14:03 -0400 Subject: [PATCH] Batch inference fixes (#8854) --- ultralytics/engine/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/predictor.py b/ultralytics/engine/predictor.py index 2282eed6..24babc73 100644 --- a/ultralytics/engine/predictor.py +++ b/ultralytics/engine/predictor.py @@ -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