相信只要是做算法改进的,首先都会遇到这么一个问题:CU,PU及TU这几个在HM中该如何打印出它们最终的划分情况呢?也经常有人来问我这个问题,一般来说,因为问我的时候我一般手头都没有现成的代码可以提供,另一方面,也没空把vs打开写上一段,所以只能是告诉他们大概的思路,其实我很早之前的一篇博客已经进行了介绍,只是当时没有把代码附上,后来也找不到了,就不了了之了。本文算是把这个遗憾给补上吧。直接附上代码,由于功能简单,就不注释了~
printf("\n---------------CTU Address(Raster Scan): %d-----------------\n", pcCU->getAddr());
printf("\n---------------CU starts--------------\n");
int iCount = 0;
int iWidthInPart = g_uiMaxCUWidth >> 2;
for (int i = 0; i < pcCU->getTotalNumPart(); i++)
{
if ( (iCount & (iWidthInPart - 1)) == 0)
printf("\n");
printf("%d ", pcCU->getDepth(g_auiRasterToZscan[i]));
iCount++;
}
printf("\n---------------CU ends--------------\n");
printf("\n---------------TU starts--------------\n");
iCount = 0;
for (int i = 0; i < pcCU->getTotalNumPart(); i++)
{
if ( (iCount & (iWidthInPart - 1)) == 0)
printf("\n");
printf("%d ", pcCU->getTransformIdx(g_auiRasterToZscan[i]) + pcCU->getDepth(g_auiRasterToZscan[i]));
iCount++;
}
printf("\n---------------TU ends--------------\n");上面这一段程序,以4x4块为单位进行打印,数字范围0~4, 0表示不划分,1表示处于深度1,依此类推。其实,只要认真研究下TComDataCU这个类,基本上你需要的那些统计信息都可以通过类似以上方法进行统计。如果想做成类似H264visa这样的分析工具,那么基于这些信息,剩下的就是图形化界面的工作了。
该博客主要针对HM编码器中CU(Coding Unit)、PU(Prediction Unit)和TU(Transform Unit)的划分问题进行讨论。作者分享了如何在HM中打印这些单元的详细划分情况,以帮助进行算法改进工作。虽然之前的文章提及过此话题但未附代码,本次博客则提供了相应的代码实现,以弥补之前的不足。

3715

被折叠的 条评论
为什么被折叠?



