From 8fd13b1b2c3a9c7e7dcdcc74c712b44b60b0ff01 Mon Sep 17 00:00:00 2001
From: shiraha721 <151541639+shiraha721@users.noreply.github.com>
Date: Tue, 11 Jun 2024 21:52:36 +0800
Subject: [PATCH] Update predictor.py
---
ultralytics/engine/predictor.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/ultralytics/engine/predictor.py b/ultralytics/engine/predictor.py
index 023b173c..637d5eb2 100644
--- a/ultralytics/engine/predictor.py
+++ b/ultralytics/engine/predictor.py
@@ -345,7 +345,8 @@ class BasePredictor:
if self.args.show:
self.show(str(p))
if self.args.save:
- self.save_predicted_images(str(self.save_dir / p.name), frame)
+
+ self.save_predicted_images(str(self.save_dir / (p.name or "tmp.jpg")), frame)
return string
@@ -375,9 +376,6 @@ class BasePredictor:
# Save images
else:
- if save_path.split("/")[-1].startswith("predict"):
- # havn't provide file name
- save_path = save_path + "/tmp.jpg"
cv2.imwrite(save_path, im)
def show(self, p=""):