aiortc视频帧PTS异常跳变问题分析与解决方案

问题背景

在使用aiortc进行音视频传输与录制时，开发者遇到了一个关于视频帧PTS(呈现时间戳)的严重问题。当网络状况不佳导致帧率下降时(低于14FPS)，接收到的视频帧PTS会出现异常跳变，数值突然增加超过40亿。这种异常导致MediaRecorder生成的视频文件时长严重失真——15分钟的实际会话被记录为超过35小时的视频。

问题现象

通过日志可以清晰地观察到PTS的异常变化：

• 正常情况下，PTS以合理增量递增(如从20148300到20239020)
• 当帧率下降时，PTS突然从20878470跳变到22596300
• 最严重时，PTS从22596300直接跳变到4317621196

问题分析

经过深入分析，发现这个问题与网络状况密切相关：

1. 当网络出现短暂中断或严重丢包时，视频帧接收间隔变大
2. aiortc内部处理机制在这种情况下会产生异常的PTS值
3. 这种跳变不是简单的数值溢出，而是某种计算错误
4. 异常PTS导致视频容器写入错误的时长信息

解决方案

经过两个月的调试，最终采用了一种相对简单但有效的解决方案：

class VideoTransformPTS_Track(MediaStreamTrack):
    """视频流轨道，用于修正异常的PTS值"""
    
    kind = "video"

    def __init__(self, track):
        super().__init__()
        self.track = track
        self._prev_frame_pts = 0  # 上一帧的原始PTS
        self._diff_pts = 0        # 两帧PTS差值
        self._show_pts = 0        # 修正后的展示PTS

    async def recv(self):
        frame = await self.track.recv()
        
        # 初始化处理
        if self._prev_frame_pts == 0:
            self._prev_frame_pts = frame.pts
            self._show_pts = frame.pts
        else:
            self._diff_pts = frame.pts - self._prev_frame_pts
            
            # 检测异常跳变(超过2秒间隔视为异常)
            if self._diff_pts > VIDEO_CLOCK_RATE * 2:
                self._show_pts = self._prev_frame_pts
                self._prev_frame_pts = frame.pts
                return None  # 丢弃异常帧
            else:
                self._prev_frame_pts = frame.pts
                frame.pts = self._show_pts + self._diff_pts
                self._show_pts = frame.pts

        return frame

实现原理

1. 跟踪PTS变化：记录每一帧的原始PTS和与前一帧的差值
2. 异常检测：当PTS增量超过2秒(VIDEO_CLOCK_RATE*2)时视为异常
3. 修正机制：
   • 对于异常帧，直接丢弃
   • 对于正常帧，基于前一帧修正后的PTS计算当前帧PTS
4. 平滑过渡：保持PTS的连续性，避免容器写入错误时长

注意事项

1. 该方案可能导致少量帧丢失，在极端网络条件下可能影响视频流畅度
2. 音频PTS未做处理，在极端情况下可能出现音视频不同步
3. VIDEO_CLOCK_RATE应根据实际视频时钟频率设置(通常为90000)

总结

这个案例展示了实时视频传输中时间戳处理的重要性。通过自定义MediaStreamTrack实现对异常PTS的检测和修正，有效解决了aiortc在弱网条件下的视频录制问题。虽然方案简单，但经过实际验证能够稳定工作，为类似场景提供了有价值的参考。