使用Python读取无人机UDP传输的JPG视频帧问题求助

你的核心问题出在UDP数据包的乱序处理和帧边界判断逻辑上，以下是具体修复步骤：

1. 关键问题点 #

• UDP是无连接协议，数据包可能因网络波动乱序到达，你直接按接收顺序拼接会破坏JPG的结构，导致图像损坏；
• 依赖数据包长度判断帧结束不可靠，正确方式是解析每个数据包的4字节头部，提取帧编号和包序号来分组排序。

2. 修复代码（两种场景适配） #

场景1：Pickle存储的是十六进制字符串 #

import pickle
from collections import defaultdict

# 读取数据包
with open("packets_data.pkl", "rb") as f:
    packet_list = pickle.load(f)

# 按帧编号分组，存储（包序号，数据）对
frames_dict = defaultdict(list)

for pckt in packet_list:
    # 提取8位十六进制头部（对应4字节）
    header_hex = pckt[:8]
    # 解析帧编号和包序号（格式需根据无人机实际协议调整，这里假设前4位帧ID，后4位包序号）
    frame_id = int(header_hex[:4], 16)
    packet_seq = int(header_hex[4:], 16)
    # 提取去掉头部后的帧数据
    data_hex = pckt[8:]
    frames_dict[frame_id].append((packet_seq, data_hex))

# 处理每个帧：排序后拼接并保存
for frame_id, packets in frames_dict.items():
    # 按包序号升序排列，确保拼接顺序正确
    sorted_packets = sorted(packets, key=lambda x: x[0])
    # 拼接所有数据
    frame_hex = ''.join([p[1] for p in sorted_packets])
    # 转字节并验证JPG格式
    frame_data = bytes.fromhex(frame_hex)
    if frame_data.startswith(b'\xff\xd8') and frame_data.endswith(b'\xff\xd9'):
        with open(f"frames/frame_{frame_id}.jpg", "wb") as f:
            f.write(frame_data)

场景2：Pickle存储的是原始字节（更常见） #

如果你的packet_list里是UDP数据包的原始字节，用这段代码：

import pickle
from collections import defaultdict

with open("packets_data.pkl", "rb") as f:
    packet_list = pickle.load(f)

frames_dict = defaultdict(list)

for pckt in packet_list:
    # 提取前4字节头部
    header = pckt[:4]
    # 解析帧ID和包序号（字节序需匹配无人机端，这里用大端）
    frame_id = int.from_bytes(header[:2], byteorder='big')
    packet_seq = int.from_bytes(header[2:], byteorder='big')
    # 提取帧数据
    data = pckt[4:]
    frames_dict[frame_id].append((packet_seq, data))

# 生成JPG帧
for frame_id, packets in frames_dict.items():
    sorted_packets = sorted(packets, key=lambda x: x[0])
    frame_data = b''.join([p[1] for p in sorted_packets])
    # 验证JPG完整性
    if frame_data.startswith(b'\xff\xd8') and frame_data.endswith(b'\xff\xd9'):
        with open(f"frames/frame_{frame_id}.jpg", "wb") as f:
            f.write(frame_data)

3. 额外验证步骤 #

1. 用Wireshark解析几个数据包的头部，确认帧编号和包序号的字节位置、字节序（大端/小端），如果和代码假设不符，调整解析逻辑；
2. 保存的帧如果仍有损坏，检查是否存在UDP丢包（可通过包序号是否连续判断），丢包导致的不完整帧直接丢弃即可。

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