【HEVC视频流可视化分析工具】画出视频中每帧的CTU块的形状与深度—

本文链接：https://blog.csdn.net/qq_47733361/article/details/149422760

文章目录

一、介绍

该【HEVC视频流可视化分析工具】的效果如下：可以画出一帧中CTU的形状或者说是深度信息，根据这附图我们可以更清晰的分析我们当前的算法对于深度的影响情况。
在这里插入图片描述

在这里插入图片描述

二、使用说明

该【HEVC视频流可视化分析工具】的第一版，需要配合日志使用，配备了完整的日志解析方法；
在使用时需要使用命令行指定日志文件，自动分析日志，提取CTU深度信息，然后生成可视化图片；

日志格式要求如下：

x265 [debug]: MODEAll: poc 0, ctu 7, All 4x4 partition depth values: 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3

MODEAll：为匹配模式，可以使用 -p 指定
“poc 0, ctu 7, All 4x4 partition depth values:”：该格式必须要有，其他没有限制

1、日志代码示例

如果只是为了看一下最终编码的CTU形状，我们可以在 source/encoder/frameencoder.cpp 中添加如下代码：

if (m_param->bCTUInfoLog) {
    std::stringstream depthStr;
    int depthCnt[4] = {0};
    for (uint32_t i = 0; i < m_param->num4x4Partitions; i++) {
        int depth = ctu->m_cuDepth[i];
        depthCnt[depth]++;
        if (i > 0) {
            depthStr << " ";  // 非首个元素前加空格分隔
        }
        depthStr << depth;
    }
    x265_log(NULL, X265_LOG_DEBUG, "MODE: poc %d, ctu %d, depthCnt %d, %d, %d, %d\n", 
            m_frame[layer]->m_poc, ctu->m_cuAddr, depthCnt[0], depthCnt[1], depthCnt[2], depthCnt[3]);
    x265_log(NULL, X265_LOG_DEBUG, "MODEAll: poc %d, ctu %d, All 4x4 partition depth values: %s\n", 
            m_frame[layer]->m_poc, ctu->m_cuAddr, depthStr.str().c_str());
}

样例：
下文样例使用x265编码器，更多的x265源码解析可以参考我的另一篇文章：【X265编码器】源代码分析——第一章编码器主干部分
compressCTU() 为压缩一个CTU，计算完成后，直接将深度depth信息输出即可；

void FrameEncoder::processRowEncoder(int intRow, ThreadLocalData& tld, int layer)
{
	......
	
	// Does all the CU analysis, returns best top level mode decision
    Mode& best = tld.analysis.compressCTU(*ctu, *m_frame[layer], m_cuGeoms[m_ctuGeomMap[cuAddr]], rowCoder);
    if (m_param->bCTUInfoLog) {
        std::stringstream depthStr;
        int depthCnt[4] = {0};
        for (uint32_t i = 0; i < m_param->num4x4Partitions; i++) {
            int depth = ctu->m_cuDepth[i];
            depthCnt[depth]++;
            if (i > 0) {
                depthStr << " ";  // 非首个元素前加空格分隔
            }
            depthStr << depth;
        }
        x265_log(NULL, X265_LOG_DEBUG, "MODE: poc %d, ctu %d, depthCnt %d, %d, %d, %d\n", 
                m_frame[layer]->m_poc, ctu->m_cuAddr, depthCnt[0], depthCnt[1], depthCnt[2], depthCnt[3]);
        x265_log(NULL, X265_LOG_DEBUG, "MODEAll: poc %d, ctu %d, All 4x4 partition depth values: %s\n", 
                m_frame[layer]->m_poc, ctu->m_cuAddr, depthStr.str().c_str());
    }
    
	......
}

三、使用样例：

python3 analysis_draw_ctu copy.py -v blue_sky_360p25-crf-new-26.ts -f blue_sky_360p25-crf-new-26.log -o output -p MODEAll

用法: analysis_draw_ctu.py -v <视频路径> -f <日志路径> [-o <输出目录>] [-p <匹配模式>]
选项:
  -v, --video:  视频文件路径（必需）
  -f, --file:   日志文件路径（必需）
  -o, --output: 输出目录（默认: output）
  -p, --pattern:日志匹配模式（默认: MODEAll）