diff --git a/backend/src/packages/chaiNNer_standard/image/video_frames/load_video.py b/backend/src/packages/chaiNNer_standard/image/video_frames/load_video.py index 1781617f4..4b51ddd26 100644 --- a/backend/src/packages/chaiNNer_standard/image/video_frames/load_video.py +++ b/backend/src/packages/chaiNNer_standard/image/video_frames/load_video.py @@ -4,7 +4,6 @@ from pathlib import Path from typing import Any -import cv2 import ffmpeg import numpy as np @@ -68,7 +67,12 @@ def load_video_node( ffmpeg_reader = ( ffmpeg.input(path) - .output("pipe:", format="rawvideo", pix_fmt="rgb24") + .output( + "pipe:", + format="rawvideo", + pix_fmt="bgr24", + sws_flags="lanczos+accurate_rnd+full_chroma_int+full_chroma_inp+bitexact", + ) .run_async(pipe_stdout=True, cmd=ffmpeg_path) ) @@ -123,7 +127,6 @@ def iterator(): print("Can't receive frame (stream end?). Exiting ...") break in_frame = np.frombuffer(in_bytes, np.uint8).reshape([height, width, 3]) - in_frame = cv2.cvtColor(in_frame, cv2.COLOR_RGB2BGR) yield in_frame, index index += 1