如何用PyAV保存接收的H.264数据包到磁盘？Pickle序列化失败

你遇到的问题本质是PyAV的av.Packet对象包含底层FFmpeg的C语言结构引用，而pickle只能序列化纯Python原生类型或实现了__reduce__接口的对象，这类带有外部资源引用的对象自然无法被正常序列化。

下面给你两种可行的解决方案：

方案1：序列化Packet的原始数据与元数据 #

我们可以把Packet中必要的信息（原始字节数据、时间戳、流索引等）提取出来，用Python原生类型（字典）保存，加载时再重新构建Packet对象。

修改后的代码如下：

import av
import time
import pickle

# 读取数据包并转换为可序列化的字典列表
container = av.open(url)  # 你的H.264流URL
packets_data = []
initial_time = time.time()

for packet in container.demux(video=0):
    # 提取Packet的关键可序列化信息
    packet_dict = {
        "data": packet.to_bytes(),
        "pts": packet.pts,
        "dts": packet.dts,
        "stream_index": packet.stream_index,
        "duration": packet.duration
    }
    packets_data.append(packet_dict)
    if time.time() - initial_time > 300:
        break

# 保存到磁盘
pickle.dump(packets_data, open("packets_data.obj", "wb"))

# 加载并重新构建Packet，解码帧
loaded_data = pickle.load(open("packets_data.obj", "rb"))
for idx, pkt_data in enumerate(loaded_data):
    # 从字节数据重建Packet
    packet = av.Packet(pkt_data["data"])
    # 恢复元数据
    packet.pts = pkt_data["pts"]
    packet.dts = pkt_data["dts"]
    packet.stream_index = pkt_data["stream_index"]
    packet.duration = pkt_data["duration"]
    
    # 解码帧并保存
    frame = packet.decode_one()
    if frame is not None:
        frame.to_image().save('frame-%04d-%04d.jpg' % (frame.index, idx))

方案2：直接保存为标准H.264文件（更推荐） #

其实没必要自己手动序列化Packet，PyAV支持直接将数据包写入标准的媒体容器文件（比如.264裸流或.ts传输流），后续读取时直接用av.open打开文件即可，这是更符合流媒体处理规范的做法。

代码示例：

import av
import time

# 读取流并写入本地H.264文件
container = av.open(url)
output_container = av.open("captured_stream.264", "w", format="h264")
# 复制输入流的参数到输出流
output_stream = output_container.add_stream(template=container.streams.video[0])

initial_time = time.time()
for packet in container.demux(video=0):
    # 调整Packet的流索引为输出流的索引
    packet.stream_index = output_stream.index
    output_container.mux(packet)
    if time.time() - initial_time > 300:
        break

# 关闭容器确保数据写入磁盘
output_container.close()
container.close()

# 后续加载并解码
load_container = av.open("captured_stream.264")
for idx, packet in enumerate(load_container.demux(video=0)):
    frame = packet.decode_one()
    if frame is not None:
        frame.to_image().save('frame-%04d-%04d.jpg' % (frame.index, idx))

这种方法的优势是不需要处理序列化细节，文件可以被其他媒体工具（比如FFmpeg、VLC）直接识别，兼容性更好。

内容的提问来源于stack exchange，提问作者Vítor Cézar