Add YUV420 encoding

dora-rs · Dec 1, 2024 · 0514044 · 0514044
1 parent 6b941f2
commit 0514044
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/node-hub/opencv-video-capture/opencv_video_capture/main.py b/node-hub/opencv-video-capture/opencv_video_capture/main.py
@@ -128,6 +128,8 @@ def main():
                 # Get the right encoding
                 if encoding == "rgb8":
                     frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                elif encoding == "yuv420":
+                    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2YUV_I420)
                 elif encoding in ["jpeg", "jpg", "jpe", "bmp", "webp", "png"]:
                     ret, frame = cv2.imencode("." + encoding, frame)
                     if not ret: