Add support for yuv[a]p16 pix_fmt

PyAV-Org · Jan 22, 2025 · 73baa3d · 73baa3d
1 parent c63aede
commit 73baa3d
Show file tree

Hide file tree

Showing 3 changed files with 197 additions and 148 deletions.
diff --git a/av/video/frame.pyi b/av/video/frame.pyi
@@ -59,15 +59,19 @@ class VideoFrame(Frame):
     ) -> VideoFrame: ...
     def to_rgb(self, **kwargs: Any) -> VideoFrame: ...
     def to_image(self, **kwargs: Any) -> Image.Image: ...
-    def to_ndarray(self, **kwargs: Any) -> _SupportedNDarray: ...
+    def to_ndarray(
+        self, force_channel_last: bool = False, **kwargs: Any
+    ) -> _SupportedNDarray: ...
     @staticmethod
     def from_image(img: Image.Image) -> VideoFrame: ...
     @staticmethod
     def from_numpy_buffer(
         array: _SupportedNDarray, format: str = "rgb24", width: int = 0
     ) -> VideoFrame: ...
     @staticmethod
-    def from_ndarray(array: _SupportedNDarray, format: str = "rgb24") -> VideoFrame: ...
+    def from_ndarray(
+        array: _SupportedNDarray, format: str = "rgb24", channel_last: bool = False
+    ) -> VideoFrame: ...
     @staticmethod
     def from_bytes(
         data: bytes,