HM中CU，TU的划分

最新推荐文章于 2022-06-24 20:22:36 发布

原创

最新推荐文章于 2022-06-24 20:22:36 发布 · 置顶 · 1.7w 阅读

33 ·

CC 4.0 BY-SA版权

文章标签：

#HEVC #CU #PU #TU #划分

该博客主要针对HM编码器中CU（Coding Unit）、PU（Prediction Unit）和TU（Transform Unit）的划分问题进行讨论。作者分享了如何在HM中打印这些单元的详细划分情况，以帮助进行算法改进工作。虽然之前的文章提及过此话题但未附代码，本次博客则提供了相应的代码实现，以弥补之前的不足。

相信只要是做算法改进的，首先都会遇到这么一个问题：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.47元/天解锁文章