avcodec_decode_video2()解码视频后丢帧的问题解决

使用libav转码视频时发现一个问题：

使用下面这段代码解码视频时，视频尾巴上会丢掉几帧。

1 2 3 4 5 6 7

while ( av_read_frame ( ifmt_ctx , & packet ) >= 0 ) {      ret = avcodec_decode_video2 ( video_dec_ctx , vframe , & got_frame , & packet ) ;      if ( got_frame ) {          packet . pts = av_rescale_q ( packet . pts , video_dec_st -> time_base , video_enc_st -> time_base ) ;          write_video_frame ( ofmt_ctx , video_enc_st , vframe ) ;      } }

这是因为源视频中PTS与DTS的不同造成的。

av_read_frame()按照PTS顺序读帧的时候，如果此帧需要参考后面的帧，那么此时avcodec_decode_video2()是没有能力解码此帧的，表现为got_frame返回0。

比如说遇上如下EFGH四帧：

ID : E F G H
KIND: I B P P
PTS : 1 2 3 4
DTS : 1 4 2 3

那么顺序读到F时，由于F需要参考G帧，而此时我们还没读到G帧，我们是没有解码F的能力的，got_frame就返回0了。如果我们对此事不做处理，那么我们就会丢掉一个帧（但丢掉的未必是F，因为av_read_frame()和avcodec_decode_video2()是1:1调用的）。

所以我们需要在while(av_read_frame())读完整个视频后，继续调用avcodec_decode_video2()把之前那些没有成功解码的帧都解出来。调用的次数就是之前got_frame返回0的次数。

按照上述思路变更代码为以下，成功找回丢失的帧。

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

int skipped_frame = 0 ;   while ( av_read_frame ( ifmt_ctx , & packet ) >= 0 ) {      ret = avcodec_decode_video2 ( video_dec_ctx , vframe , & got_frame , & packet ) ;      if ( got_frame ) {          packet . pts = av_rescale_q ( packet . pts , video_dec_st -> time_base , video_enc_st -> time_base ) ;          write_video_frame ( ofmt_ctx , video_enc_st , vframe ) ;      }      else      {          skipped_frame ++ ;      } }   for ( int i = skipped_frame ; i > 0 ; i -- ) {      ret = avcodec_decode_video2 ( video_dec_ctx , vframe , & got_frame , & packet ) ;      if ( got_frame ) {          packet . pts = av_rescale_q ( packet . pts , video_dec_st -> time_base , video_enc_st -> time_base ) ;          write_video_frame ( ofmt_ctx , video_enc_st , vframe ) ;      } }

anyShare

转自： http://blog.yikuyiku.com/?p=3486