From 38bcce53c216f512e0e4a1b51a132ee35aef6eb7 Mon Sep 17 00:00:00 2001
From: shiraha721 <151541639+shiraha721@users.noreply.github.com>
Date: Tue, 11 Jun 2024 20:48:47 +0800
Subject: [PATCH] fix cv2.error when use PIL.Image.Image

---
 ultralytics/engine/predictor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ultralytics/engine/predictor.py b/ultralytics/engine/predictor.py
index 921f273f..023b173c 100644
--- a/ultralytics/engine/predictor.py
+++ b/ultralytics/engine/predictor.py
@@ -375,6 +375,9 @@ 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=""):