From 3a53d91ec833c99d4c6215f3a67582dd2f2172a4 Mon Sep 17 00:00:00 2001
From: shiraha721 <151541639+shiraha721@users.noreply.github.com>
Date: Tue, 11 Jun 2024 21:55:42 +0800
Subject: [PATCH] fix cv2.error when use PIL.Image.Image (#241)

---
 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 921f273f..9ec803a7 100644
--- a/ultralytics/engine/predictor.py
+++ b/ultralytics/engine/predictor.py
@@ -345,7 +345,7 @@ 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