VVC学习之二：VTM中CU划分结构QTMTT（3）：打印QTMTT最终划分

介绍完QTMTT在VTM中的实现，相信很多人包括我也对QTMTT的最终划分结果很感兴趣，于是我尝试将最终的划分结果进行可视化显示。

在xCompressCU中，将对CTU进行递归划分，用RDcost选择出最优的划分方式，因此在xCompressCU中很难提取到QTMTT的最优划分方式。因此，要得到最终划分结果，必须在xCompressCU之后。VTM提供了遍历每个CodingStructure中所有CU的函数traverseCUs()，因此只需要在xCompressCU之后，遍历CTU中每个CU就可以得到其位置和尺寸信息。

为了可视化显示CU，需要修改CU边界像素值，但是重建像素将会用作参考，直接修改重建值将会影响帧间预测的运动估计和运动补偿，因此不能直接修改CU的重建值。我用了最笨的方法，copy重建帧，在copy上进行可视化，并保存下来。

下面给出实现过程

// 遍历每帧图像中所有CTU，对每个CTU遍历所有CU
for (int y = 0; y < pcv.heightInCtus; y++){for (int x = 0; x < pcv.widthInCtus; x++){const UnitArea ctuArea(pcv.chrFormat, Area(x << pcv.maxCUWidthLog2, y << pcv.maxCUHeightLog2, pcv.maxCUWidth, pcv.maxCUWidth));//  CU-based paintingif (CS::isDualITree(cs)){for (auto &currCU : cs.traverseCUs(CS::getArea(cs, ctuArea, CH_L), CH_L)){xDrawCUIntraColor(currCU, Height, Width, pbufY, pbufU, pbufV);}}else{for (auto &currCU : cs.traverseCUs(CS::getArea(cs, ctuArea, CH_L), CH_L)){xDrawCUColor(currCU, Height, Width, pbufY, pbufU, pbufV);}}}}

// 这里根据每个CU的编码模式，选择了不同的显示颜色
void xDrawCUIntraColor(CodingUnit& cu, int orgHeight, int orgWidth, unsigned char* tempY, unsigned char* tempU, unsigned char* tempV)
{const CompArea&  lumaArea = cu.block(COMPONENT_Y);int cuX = lumaArea.x;int cuY = lumaArea.y;int cuH = lumaArea.height;int cuW = lumaArea.width;int chromaX = cuX >> 1;int chromaY = cuY >> 1;int chromaH = cuH >> 1;int chromaW = cuW >> 1;unsigned char y = 0, u = 0, v = 0;if (cu.firstPU->intraDir[CHANNEL_TYPE_LUMA] < 2){rgb2yuv(0, 0, 255, y, u, v);      // blue}else{rgb2yuv(255, 0, 0, y, u, v);      // red}int tempLumaIdx;for (int i = 0; i < cuH; i++){for (int j = 0; j < cuW; j++){tempLumaIdx = (i + cuY) * orgWidth + cuX + j;if (i == cuH - 1 || j == cuW - 1) //i == 0 || i == cuH - 1 || j == 0 || j == cuW - 1{tempY[tempLumaIdx] = y;}}}int tempChromaIdx;for (int i = 0; i < chromaH; i++){for (int j = 0; j < chromaW; j++){tempChromaIdx = (i + chromaY) * orgWidth / 2 + (chromaX + j);if (i == chromaH - 1|| j == chromaW - 1)  //i == 0 || i == chromaH - 1 || j == 0 || j == chromaW - 1{tempU[tempChromaIdx] = u;tempV[tempChromaIdx] = v;}}}
}

最终的效果如下：

帧内划分结果

帧间划分结果

VVC学习之二：VTM中CU划分结构QTMTT（3）：打印QTMTT最终划分相关推荐

VVC学习之四：VTM中的数据结构——CodingUnit、PredictionUnit、TransformUnit
文章目录 CodingUnit PredictionUnit TransformUnit 基本单元的结构体中,只定义了CU.PU.TU的属性信息,其头文件为Unit.h,对基本单元的操作定义在Unit ...
（转）SpringMVC学习(十二)——SpringMVC中的拦截器
http://blog.csdn.net/yerenyuan_pku/article/details/72567761 SpringMVC的处理器拦截器类似于Servlet开发中的过滤器Filter, ...
VVC学习之五：VTM帧内预测工具详解
文章目录简介 1. 扩展角度帧内模式 1.1 MPM构建和模式编码 1.2 帧内广角预测(WAIP) 1.3 基于模式的参考样本平滑(MDIS) 2. 跨分量线性模型预测(CCLM) 3. 位置自适 ...
JavaWeb学习总结(二)——JSP中的九个内置对象
一.JSP运行原理每个JSP 页面在第一次被访问时,WEB容器都会把请求交给JSP引擎(即一个Java程序)去处理.JSP引擎先将JSP翻译成一个_jspServlet(实质上也是一个servlet ...
【theano-windows】学习笔记二——theano中的函数和共享参数
前言上一篇博客中学到了theano中的变量类型,也就是dscalar.dvector之类的, 然后还有一个theano.function和eval函数, 将我们所定义的操作转换成theano可执行的 ...
CNS文章代码学习（二）Immunity 三级淋巴结构文章学习
前言:经典的文章值得反复学习,正好也有同学提问figure3的问题:因为我也不是专业的生信研究生,所以只能浅浅讲一讲我的想法和理解. Summary: 肿瘤内三级淋巴结构TLS的出现与良好的临床预后和 ...
C++虚幻-密室逃脱学习总结(二)虚幻4文件夹结构及基本操作
文件夹结构打开你所下载的虚幻引擎首先大致了解下目录结构,进入你存储的路径找到项目名文件夹下会有如下几个文件夹: 以下是官方文档中解释的目录存放内容官方文档地址(建议保存):http://api. ...
Python学习篇(五) Python中的循环
文章目录前言一.range函数二.while循环 2.1四步循环法三.for in 循环四.流程控制语句 4.1 break 4.2 continue 五.else语句六.嵌套循环七.二 ...
storm目录结构及在zk中的目录结构
storm目录结构及在zk中的目录结构 @(STORM)[storm] storm目录结构及在zk中的目录结构一storm在磁盘中的内容一nimbus中的目录结构 1inbox目录 2stormd ...

VVC学习之二：VTM中CU划分结构QTMTT（3）：打印QTMTT最终划分

VVC学习之二：VTM中CU划分结构QTMTT（3）：打印QTMTT最终划分相关推荐

最新文章

热门文章