fix cv2.error when use PIL.Image.Image

2025-05-23 05:24:22 +08:00 · 2024-06-11 20:48:47 +08:00 · 2024-06-11 20:48:47 +08:00 · 38bcce53c2
commit 38bcce53c2
parent 0efdd2f0c4
1 changed files with 3 additions and 0 deletions
--- 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=""):