Raise an exception if cv2 fails to read an image path

PTG-Kitware · Dec 18, 2024 · f857f8f · f857f8f
1 parent 75d3d27
commit f857f8f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/angel_system/object_detection/yolov11_detect.py b/angel_system/object_detection/yolov11_detect.py
@@ -337,6 +337,8 @@ def yolo_v11_inference_objects(
     ):
         img_path: Path = dset.get_image_fpath(img_id)
         img0 = cv2.imread(img_path.as_posix())
+        if img0 is None:
+            raise RuntimeError(f"Failed to read image file: {img_path}")
 
         # returns list of length=num images, which is always 1 here.
         object_preds = object_model.predict(source=img0, **object_predict_kwargs)[0]