From 12de41521c60bb2ee3149816e729b7b20f8e2b52 Mon Sep 17 00:00:00 2001 From: NevermindNilas Date: Tue, 31 Dec 2024 01:51:50 +0200 Subject: [PATCH] make sure to force hwc --- src/utils/ffmpegSettings.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/utils/ffmpegSettings.py b/src/utils/ffmpegSettings.py index 0f4a33f5..503e9d26 100644 --- a/src/utils/ffmpegSettings.py +++ b/src/utils/ffmpegSettings.py @@ -453,6 +453,7 @@ def __init__( device="cpu", num_threads=decodeThreads, filters=filters, + tensor_shape="HWC", )([float(inpoint), float(outpoint)]) else: self.reader = VideoReader( @@ -460,6 +461,7 @@ def __init__( device="cpu", num_threads=decodeThreads, filters=filters, + tensor_shape="HWC", ) logging.info("Using Celux pipeline for video decoding") except Exception as e: