且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

h264检测是I帧还是P帧,解决录像第一帧有马赛克问题

更新时间:2022-09-16 19:46:37

最近使用h264码流数据进行录像,但是录出来的第一帧有马赛克,究其原因是录像的第一帧不是关键帧,所以需要录像是需要判断第一帧是否是关键帧,方法有两种,第一种是原码流的基础上查找,第二种是将原码流传递给ffmpeg让ffmpeg判断key_frame,第二种相对简单,但是由于这样一来录像和解码视频切合的太紧密,之后修改比较麻烦,所以选择第一种方案,查看注释1可以明白如何检测h264码流的关键帧,下面是我截取原码流的关键帧和p帧

 

0000000 :是sps

0000000  :  是pps

0000000  : 是帧类型

关键帧类型:

 

[cpp] view plaincopy
 
  1. <span style="color:#FF0000;"><span style="font-size:24px;">0000000: <span style="color:#CCCCCC;">0000 0001 6742 401f 9654 0501 ed00 f39e  ....gB@..T......  
  2. 0000010: a</span><span style="color:#3366FF;">000 0000 0168 ce38 80</span>00 0000 01<span style="color:#000000;">65</span> 8880  .....h.8.....e..  
  3. 0000020: 4001 8231 2000 4f11 d84d 5fff fb3b c28a  @..1 .O..M_..;..  
  4. 0000030: 00bc fc83 03db b3e3 8603 9c59 fa0f a82c  ...........Y...,  
  5. 0000040: df55 fdf6 8414 032a e766 bd4b fbea 05af  .U.....*.f.K....</span>  
  6. </span>  



 

P帧类型:

 

[cpp] view plaincopy
 
  1. 0000000: <span style="color:#C0C0C0;">0000 0001 6742 401f 9654 0501 ed00 f39e  ....gB@..T......  
  2. 0000010: a</span><span style="color:#3366FF;">000 0000 0168 ce38 80</span>00 0000 01<span style="color:#000000;">41</span> 9a02  .....h.8.....A..  
  3. 0000020: 0586 7cb9 9125 5788 8f90 7f1f 1930 7eef  ..|..%W......0~.  
  4. 0000030: 6383 bebd 2cc5 3627 92c3 390b 46dc d4a5  c...,.6'..9.F...  
  5. 0000040: 774b 3484 57f8 9840 fba3 1dd6 800f 2242  wK4.W..@......"B  
  6. 0000050: 8816 080f 8f8d 84c6 09aa cda6 363d 00da  ............6=..  
  7. 0000060: b563 4392 bc65 93e2 63bb 6d30 472e 3ef1  .cC..e..c.m0G.>.  
  8. 0000070: 545d 6a3f 36c3 2f7d 6b1e 3c91 d15d d687  T]j?6./}k.<..]..  


所以在代码中需要检索第29个字节,来判断是65还是41,

 

 

[python] view plaincopy
 
  1. public static String byteToHexString(byte src){     
  2.         StringBuilder stringBuilder = new StringBuilder("");     
  3.             int v = src & 0xFF;     
  4.             String hv = Integer.toHexString(v);    
  5.             if (hv.length() < 2) {     
  6.                 stringBuilder.append(0);   
  7.             }     
  8.             stringBuilder.append(hv);  
  9.         return stringBuilder.toString();     
  10.     }     
  11. private boolean isFirstIFrame = true;  
  12. private String IFrame = "65";//关键帧是0x65  
  13. //<span style="font-size:16px"></span>bArrayImage是存放h264原码流字节数组  
  14. if (是在录像) {  
  15.                 if (isFirstIFrame) {  
  16.                     String type = byteToHexString(bArrayImage[29]);  
  17.                     if (type.equals(IFrame)) {    //第29个字符是判断帧的类型  
  18.                         isFirstIFrame = false;  
  19.                         ShootingVideoData(bArrayImage, Video_Data_iVideoLen);//录制第一帧:关键帧  
  20.                     }  
  21.                 }else {  
  22.                     ShootingVideoData(bArrayImage, Video_Data_iVideoLen);  
  23.                 }  
  24.             }  

这样在录制出的第一帧录像就没有马赛克了。

 

注释1:检测h264中I帧,P帧

原文链接:http://blog.csdn.net/zgyulongfei/article/details/7558031

今天在网上找了一些资料,知道了如何检测h264中的帧类型,在这里记录下来。

首先,贴出nal单元类型定义(图从《新一代视频压缩编码标准H.264》摘录):

h264检测是I帧还是P帧,解决录像第一帧有马赛克问题

 

假设一段h264的码流为:00 00 00 01 41 E6 60……

其中的00 00 00 01为起始码,而起始码之后的下一个字节就可以检测出这一帧的类型。

在上面的码流中起始码之后的字节位 0x41,换算成二进制为 0100 0001。

注:我解读顺序为从左往右算。

(1)第1位禁止位,值为1表示语法出错

(2)第2~3位为参考级别

(3)第4~8为是nal单元类型,在此处为 0 0001换算成十进制为1。根据上图可知道这段码流是【不分区、非IDR图像的片】,在baseline的档次中就是P帧,因为baseline没有B帧。

如果是另一段码流:00 00 00 01 65 E8……

那么根据0x65字节(0110 0101)根据后5位换算十进制为5,也就是【IDR图像中的片】,即I帧。

 

用代码的方式可以这样写,int type = 0x65 & 0x1f,然后根据type在表中查找即可获得需要的结果。