Fix pyav 14 error

pytorch · Dec 2, 2024 · 634b95c · 634b95c
1 parent 36e219b
commit 634b95c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/torchvision/io/video.py b/torchvision/io/video.py
@@ -155,7 +155,13 @@ def write_video(
 
         for img in video_array:
             frame = av.VideoFrame.from_ndarray(img, format="rgb24")
-            frame.pict_type = "NONE"
+            try:
+                frame.pict_type = "NONE"
+            except TypeError:
+                from av.video.frame import PictureType  # noqa
+
+                frame.pict_type = PictureType.NONE
+
             for packet in stream.encode(frame):
                 container.mux(packet)