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=""):