Check if image exists in read_plate_image

Author: ankandrew

fast_plate_ocr/inference/process.py

@@ -2,6 +2,8 @@
 Utility functions for processing model input/output.
 """
 
+import os
+
 import cv2
 import numpy as np
 import numpy.typing as npt
@@ -14,6 +16,8 @@ def read_plate_image(image_path: str) -> npt.NDArray:
     :param str image_path: The path to the license plate image.
     :return: The image as a NumPy array.
     """
+    if os.path.exists(image_path):
+        raise ValueError(f"{image_path} file doesn't exist!")
     return cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)