Skip image that cannot be encoded

dora-rs · Oct 10, 2024 · 254e74d · 254e74d
1 parent ce408a1
commit 254e74d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/node-hub/opencv-video-capture/opencv_video_capture/main.py b/node-hub/opencv-video-capture/opencv_video_capture/main.py
@@ -120,7 +120,10 @@ def main():
                 if encoding == "rgb8":
                     frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
                 elif encoding in ["jpeg", "jpg", "jpe", "bmp", "webp", "png"]:
-                    frame = cv2.imencode("." + encoding, frame)[1]
+                    ret, frame = cv2.imencode("." + encoding, frame)
+                    if not ret:
+                        print("Could not encode image...")
+                        continue
 
                 storage = pa.array(frame.ravel())