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