[发明专利]编码单元划分方法、图像编解码方法和装置

说明书

本发明实施例提供一种编码单元划分方法、图像编解码方法和装置，该编码单元划分方法包括：对待划分的处理单元进行下采样，以得到预定大小的待划分块；将该待划分块转换得到的第一向量输入至训练后的神经网络模型中，获取该神经网络模型的输出结果，该输出结果包括水平方向划分概率和垂直方向划分概率；在该水平方向划分概率大于或等于阈值时，对该待划分的处理单元进行水平划分，在该垂直方向划分概率大于或等于该阈值时，对该待划分的处理单元进行垂直划分。

技术领域

本发明涉及图像处理领域。

背景技术

多功能视频编码(Versatile Video Coding，VVC)是由联合视频专家团队(JointVideo Experts Team，JVET)建立的一项新的标准化项目，其主要目标是改进现有高效视频编码(High Efficiency Video Coding，HEVC)标准，提供更高的压缩性能。目前，在全帧内(All Intra，AI)配置中，VVC相较于HEVC提升了约25％的编码效率，但编码时间增加了10倍

应该注意，上面对技术背景的介绍只是为了方便对本发明的技术方案进行清楚、完整的说明，并方便本领域技术人员的理解而阐述的。不能仅仅因为这些方案在本发明的背景技术部分进行了阐述而认为上述技术方案为本领域技术人员所公知。

发明内容

在VVC中，使用了比HEVC更复杂的编码单元划分结构，在HEVC四叉树(QT)划分的基础上增加了两种二叉树(BT)划分和两种三叉树(TT)划分(二叉树和三叉树统称为多类型树MTT)，图1A-图1B是该两种二叉树划分示意图，分别为垂直方向的二叉树划分和水平方向的二叉树划分，图2A-图2B是该两种三叉树划分示意图，分别为垂直方向的三叉树划分和水平方向的三叉树划分，在编码时，图像将被划分成一系列编码树单元(CTU，coding treeunit)，针对每个CTU，先进行四叉树划分；接着，对经过四叉树划分后的各个单元进行独立地或相同地MTT划分，划分后得到的这些节点称为编码单元(CU，coding unit)，在图像编解码时，可以以CU为单位进行预测、变换等处理。另外每一个CTU包含亮度分量和色度分量，针对各个亮度和色度分量可以使用相同的划分结构或不同的划分结构。

发明人发现，在现有方法中，对经过四叉树划分后的各个单元，将依次遍历各种可行的划分结构，计算不同的划分结构下的编码结果对应的率失真函数值，并根据率失真函数值从中选择最佳编码性能对应的划分结构，这无疑增加了编码单元划分的复杂度，增加了编码的时间。

针对上述问题中的至少之一，本发明实施例提供一种编码单元划分方法、图像编解码方法和装置。

根据本发明实施例的第一方面，提供一种编码单元划分装置，其中，该装置包括：

采样模块，其用于对待划分的处理单元进行下采样，以得到预定大小的待划分块；

获取模块，其用于将该待划分块转换得到的第一向量输入至训练后的神经网络模型中，获取该神经网络模型的输出结果，该输出结果包括水平方向划分概率和垂直方向划分概率；

确定模块，其用于在该水平方向划分概率大于或等于阈值时，对该待划分的处理单元进行水平划分，在该垂直方向划分概率大于或等于阈值时，对该待划分的处理单元进行垂直划分。

根据本发明实施例的第二方面，提供一种图像编解码装置，其中，该装置包括：

划分模块、第一方面该的编码单元划分装置以及编解码模块；

该划分模块用于将图像划分为多个待划分的处理单元；