hevc参考代码matlab版,HEVC部分源码剖析

HEVC(High Efficiency Video Coding) 是2013年提出的最新的视频编码标准,它的核心是将视频中的每一帧图像分割成 Coding Tree Unit(CTU) ,然后对每个CTU确定最佳的分割深度,关于具体的算法可以参考:

在了解了一些HEVC内部的术语如CTU, CU, TU...后,我们来试着利用HEVC的源代码输出CTU的分割信息。

代码的下载和配置

首先,需要下载源代码,源代码是使用SVN托管的,因此需要用SVN下载,我们可以使用TortoiseSVN进行下载。安装好TortoiseSVN后,打开SVN browser,在地址栏输入

就可以看到源代码的目录结构,然后选择export,就可以将源代码保存到本地。注意这个目录下面有很多个版本,我们这里选择/trunk/目录下的主版本导出。

导出到本地之后,进入/build/文件夹,里面就是编译所需的文件,支持不同的平台:

b8d61f3899a5

compilation

我使用的是VS2015编译,打开这个工程之后,可以看到所有的项目,对应的源代码在/source/文件夹下:

b8d61f3899a5

HEVC projects

选择F7或者Build Solution就可以进行编译,生成exe文件,但是如果你直接去运行的话会发现程序输出一段说明文字之后就结束了,这是因为我们需要先进行配置,指定输入的视频和其他的一些参数。

配置文件在/cfg/目录下,其中最主要的两个配置文件是bitstream.cfg和encoder_intra_main.cfg,我们把这两个文件复制到一个自定义工作目录下,比如E:\HM\trunk\workspace,接下来在Visual Studio中,右击解决方案中“TAppEncoder”->“设为启动项目”

再右击“TAppEncoder”->”属性”->”配置属性”->”调试” ,在工作目录栏指定工作目录路径E:\HM\trunk\workspace,在命令参数栏中填写-c encoder_intra_main.cfg -c bitstream.cfg,如图:

b8d61f3899a5

config

对于encoder_intra_main.cfg,我们需要进行配置的参数就只有:

#======== Quantization =============

QP : 32 # Quantization parameter(0-51)

这个值从0到51都可以。而在bitstream.cfg中,我们需要配置以下的参数:

b8d61f3899a5

bitstream.cfg

InputFile后面写上输入的.YUV视频文件的绝对路径,FramesToBeEncoded是指你想编码多少帧,至于FrameRate, SourceWidth, SourceHeight这些信息都在视频文件名中,比如我这里使用的BasketballDrill_832x480_50.yuv,文件名中就能看出宽度、高度和帧率。

接下来就可以运行了,运行结束后,工作目录下会出现两个文件:str.bin和rec.yuv,其中rec.yuv是编码过程中重建的yuv图像,str.bin则是压缩后的码流。

YUV文件

首先提供一些YUV视频资源的下载地址:

对于HEVC的测试输入,一般都是采用YUV文件,推荐安装一个YUV播放器:YUView,可以看到每一帧的图片。

YUV文件就是由一帧一帧的图像构成的,在之前提到的配置文件中的参数FramesToBeEncoded中,你指定的编码多少帧,HEVC就会取前面的多少帧进行编码。如果想提取出YUV文件中的每一帧,可以使用FFmpeg:

ffmpeg -video_size 832x480 -r 50 -pixel_format yuv420p -i BasketballDrill_832x480_50.yuv output-%d.png

当然,使用的时候这个命令中的视频大小、帧率、像素格式、视频名称都需要根据具体的视频修改。其中像素格式(-pixel_format)这个参数可能很多人不知道该选什么,在命令行中输入:

ffmpeg -pix_fmts

可以看到FFmpeg支持的格式,这里使用的yuv420p是一般采用的格式,420代表的是 YCbCr color space with 4:2:0 sampling.

This separates a color representation into three components called Y, Cb, and Cr. The Y component is also called luma, and represents brightness. The two chroma components Cb and Cr represent the extent to which the color deviates from gray toward blue and red, respectively.

关于color space,可以参考:

colorspace – FFmpeg

如果不想了解这么多,只想直接知道自己所使用的YUV应该对应哪种 pixel format,方法是使用YUView打开YUV文件,在右侧的properties中有一个YUV Format,然后用这个format,去对应:

Most commonly used formats

pixel_format

8-bit 4:2:0

yuv420p

8-bit 4:2:2

yuv422p

8-bit 4:4:4

yuv444p

10-bit 4:2:0

yuv420p10le

10-bit 4:2:2

yuv422p10le

CTU分割信息的输出

这一部分是在Encoder部分完成的,所以我们主要看TAppEncoder这个项目。在TEncGOP.cpp中,有一个precompressSlice()和一个compressSlice()函数,前者可以不用管,而后者则计算出了每个ctu的最佳分割深度,转到compressSlice()函数的定义,来到了TEncSlice.cpp,其中对ctu进行最佳深度计算的关键函数是compressCtu()这个函数。

在compressCtu()这个函数内部,当xCompressCU()这个函数运行完成之后,最佳的分割深度就已经得到了,此时,如果想要输出当前ctu最佳的分割,可以在xCompressCU()这个函数后面,加上下面的语句:

xCompressCU( m_ppcBestCU[0], m_ppcTempCU[0], 0 DEBUG_STRING_PASS_INTO(sDebug) );

//============== add code from here ===============

TComDataCU* DepthCU = m_ppcBestCU[0];

UInt tempDepth;

ofstream outfile("PartitionInfo.txt", ios::in | ios::app);

for (UInt iPartitionNum = 0; iPartitionNum < DepthCU->getTotalNumPart(); iPartitionNum++)

{

if (iPartitionNum % 16 == 0) {

outfile << endl;

}

tempDepth = DepthCU->getDepth(g_auiRasterToZscan[iPartitionNum]);

outfile << " " << tempDepth;

}

outfile << endl;

// =============code added end here===============

最佳的分割保存在m_ppcBestCU[0]里面,使用getDepth()[i]就可以得到。

接下来,如果我们不仅想输出所有的ctu分割信息,还想输出当前是第几个ctu,这时候就要去找哪个变量记录了当前ctu的编号,我们可以回到compressCtu()函数开始的地方,看到这个函数接收了一个参数TComDataCU* pCtu,而我们可以通过pCtu->getCtuRsAddr()这个语句得到当前CTU的编号,所以,可以在xCompressCU()之前,加上下面的语句:

//============== add code from here ===============

ofstream outfile("PartitionInfo.txt", ios::in | ios::app);

UInt temp_ctu_addr = pCtu->getCtuRsAddr();

outfile << "ctu:" << temp_ctu_addr << endl;

outfile.close();

// =============code added end here===============

xCompressCU(m_ppcBestCU[0], m_ppcTempCU[0], 0 DEBUG_STRING_PASS_INTO(sDebug));

除了输出当前ctu的编号,我们还想要输出当前编码到了第几帧(frame),这时仍然可以通过Debug的方式找到存储这一信息的变量。最终发现当前编码的frame数量存储在变量m_iFrameRcvd中,而这一变量是在TAppEncTop.cpp文件中。因此,如果要输出当前是第几帧,可以在TAppEncTop.cpp文件中找到

m_iFrameRcvd++;

这一语句,并在后面加入:

ofstream outfile("PartitionInfo.txt", ios::in | ios::app);

outfile<< "frame:" << m_iFrameRcvd << endl;

这样,我们就可以输出每个frame对应的每个ctu的分割信息了,像这样:

b8d61f3899a5

PartitionInfo

然后我们可以使用Matlab画出每一帧的PU分割信息,参考:

最后补充一下,HEVC在帧内预测的时候,对于每个depth,都要向下分成4个depth递归寻找最优分割,因此我们还可以把每个depth对应的RD-cost打印出来。Depth0会分成4个Depth1,Depth1会分成4个Depth2,Depth2分成4个Depth3,Depth3向下判断是8x8还是4x4的PU

打印出每种分割的RD-cost的好处就是,当我们训练好了神经网络模型,用模型生成CU分割深度信息的时候,我们可以用RD-cost来和HEVC编码器比较RD-cost变化了多少,而神经网络模型得到的分割方法的RD-cost的计算就可以用之前打印出来的每个depth的RD-cost

在xCompressCU()函数内部,有两个变量:m_ppcBestCU和m_ppcTempCU,想要得到当前深度的RD-cost可以通过调用它们的getTotalCost()方法,不过m_ppcBestCU只存放了最佳的分割深度,所以不能得到全面的RD-cost,因此只能调用m_ppcTempCU的getTotalCost()方法。但m_ppcTempCU在xCompressCU()函数内部并没有存放或者被赋值RD-cost信息,我们只能到xCheckRDCostIntra()函数内部来获取。

在xCheckRDCostIntra()函数内部,当前所在的深度存放在变量uiDepth中。我们可以在

rpcTempCU->getTotalCost() = m_pcRdCost->calcRdCost( rpcTempCU->getTotalBits(), rpcTempCU->getTotalDistortion() );

之后,添加如下代码,输出每一层深度的RD-cost:

rpcTempCU->getTotalBits() = m_pcEntropyCoder->getNumberOfWrittenBits();

rpcTempCU->getTotalBins() = ((TEncBinCABAC *)((TEncSbac*)m_pcEntropyCoder->m_pcEntropyCoderIf)->getEncBinIf())->getBinsCoded();

rpcTempCU->getTotalCost() = m_pcRdCost->calcRdCost( rpcTempCU->getTotalBits(), rpcTempCU->getTotalDistortion() );

// ==============added code to print depth and RDcost==========

ofstream outfile("rdcost.txt", ios::in | ios::app);

double temp_rdcost;

temp_rdcost = rpcTempCU->getTotalCost();

outfile << "depth:" << uiDepth << endl;

outfile << temp_rdcost << endl;

outfile.close();

// ===============code added end here=====================

xCheckDQP( rpcTempCU );

同样,我们还是需要输出是在哪一帧,以及哪个CTU。输出之后,在进行一些处理,就能方便地计算自己的模型的CU划分对应的RD-cost了。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值