[发明专利]图像划分方法和装置

权利要求书

1.一种图像划分方法，其特征在于，包括：

获得图像中待划分的编码树单元CTU；

将所述CTU输入到划分深度识别模型，得到所述划分深度识别模型预测出的所述CTU的划分深度特征，所述CTU的划分深度特征包括所述CTU中各基本单元对应的划分深度，所述CTU包括K*K个互不重叠的基本单元，且每个基本单元由L*L个像素点构成，其中，K＝2ⁿ，n为大于等于1的自然数，L为大于等于2的自然数；其中，所述划分深度识别模型为利用标注有实际划分深度特征的多个CTU样本训练卷积神经网络模型得到的，所述CTU样本的实际划分深度特征为基于率失真优化方式确定出的所述CTU样本中各个基本单元对应的划分深度；

基于所述CTU中各基本单元的划分深度，确定所述CTU所需划分出的至少一个编码单元CU，并将所述CTU划分为所述至少一个编码单元CU。

2.根据权利要求1所述的方法，其特征在于，所述基于所述CTU中各基本单元的划分深度，确定所述CTU所需划分出的至少一个编码单元CU，并将所述CTU划分为所述至少一个编码单元CU，包括：

依据CTU中各所述基本单元的划分深度以及不同划分深度的编码单元CU所需包含的基本单元的数量，并按照所述CTU中各基本单元从左到右的第一顺序以及从上到下的第二顺序，依次确定出CTU中各基本单元所属的CU，并将CTU划分为确定出的至少一个编码单元。

3.根据权利要求2所述的方法，其特征在于，所述依据CTU中各所述基本单元的划分深度以及不同划分深度的编码单元CU所需包含的基本单元的数量，并按照所述CTU中各基本单元从左到右的第一顺序以及从上到下的第二顺序，依次确定出CTU中各基本单元所属的CU，并将CTU划分为确定出的至少一个编码单元，包括：

按照所述CTU中各基本单元从左到右的第一顺序以及从上到下的第二顺序，确定出所述CTU中尚未确定所归属的编码单元的目标基本单元；

如所述目标基本单元的划分深度为0，则将所述CTU中所有基本单元确定为归属于一个编码单元，得到一个具有T1*T1个像素点的第一编码单元，T1＝K*L；

如所述目标基本单元的划分深度为i，以所述目标基本单元为基准，并按照所述第一顺序和第二顺序，确定出包含所述目标基本单元在内的个基本单元，将所述个基本单元确定为一个具有T2*T2个像素点的第二编码单元，并从所述CTU中划分出所述第二编码单元，其中，i为从1到n-1的自然数；

如所述目标基本单元的划分深度为n，将所述目标基本单元确定为一个编码单元，得到具有L*L个像素点的第三编码单元，并从所述CTU中划分出所述第三编码单元；

如所述CTU中仍存在尚未确定所归属的编码单元的基本单元，则返回执行所述确定出所述CTU中尚未确定所归属的编码单元的目标基本单元的操作，直至所述CTU中不存在尚未确定所归属的编码单元的基本单元，得到所述CTU划分出的至少一个编码单元CU。

4.根据权利要求1所述的方法，其特征在于，在所述基于所述CTU中各基本单元的划分深度，确定所述CTU所需划分出的至少一个编码单元CU之前，还包括：

按照设定的符合四叉树划分规范的深度分布规律，确定所述CTU中划分深度存在异常的异常基本单元，并对所述异常基本单元的划分深度进行修正。