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())