使用PyAV的av.open()读取BytesIO对象失败如何解决

问题原因 #

出现流索引为随机乱值、无法正常解码帧的问题，是两个常见踩坑点导致的：

• 读取本地文件时，PyAV可以通过文件名后缀、文件元数据自动识别视频容器格式，解析流程稳定；但传入BytesIO这类内存文件对象时，默认的自动格式探测逻辑不稳定，很容易解析失败，此时流对象里的索引字段没有被正确赋值，读到的是内存残留的垃圾值，就会出现0、1054153136这类随机数。
• 如果传入的BytesIO对象读写指针不在字节流起始位置（比如之前做过读写操作没有重置指针），PyAV会从指针当前位置开始读取字节，拿不到完整的mp4文件头，自然解析出的元数据完全错乱。

修复代码 #

不需要把字节落盘存临时文件，两步修改就能解决：

1. 调用av.open前先把BytesIO指针重置到字节流开头
2. 显式指定视频容器格式，跳过不稳定的自动探测逻辑

import io
import av

some_bytesio_object = io.BytesIO(some_byte_file)
# 重置指针到流起始位置
some_bytesio_object.seek(0)
# 显式指定容器格式为mp4，和实际字节对应的视频格式保持一致
container = av.open(some_bytesio_object, format="mp4")

print(container.streams.video)
# 此时会正常输出 <av.VideoStream #0 h264, yuv420p 1280x720 ...> 格式的正确信息
for frame in container.decode(video=0):
    frame.to_image().save('frame-{}.jpg'.format(frame.index))
# 可以正常导出所有视频帧

注意事项 #

• 如果你的视频不是mp4格式，替换format参数为对应容器的FFmpeg格式标识即可：flv对应"flv"、mkv对应"matroska"、ts对应"mpegts"。
• 不要依赖内存流的自动格式探测，该逻辑在PyAV多个历史版本中都存在兼容性问题，显式指定格式是最稳定的方案。

内容的提问来源于stack exchange，提问作者Smith