make sure to force hwc

NevermindNilas · Dec 30, 2024 · 12de415 · 12de415
1 parent 3760628
commit 12de415
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/utils/ffmpegSettings.py b/src/utils/ffmpegSettings.py
@@ -453,13 +453,15 @@ def __init__(
                         device="cpu",
                         num_threads=decodeThreads,
                         filters=filters,
+                        tensor_shape="HWC",
                     )([float(inpoint), float(outpoint)])
                 else:
                     self.reader = VideoReader(
                         videoInput,
                         device="cpu",
                         num_threads=decodeThreads,
                         filters=filters,
+                        tensor_shape="HWC",
                     )
                 logging.info("Using Celux pipeline for video decoding")
             except Exception as e: