[發明專利]圖像劃分方法和裝置

在審

• 申請號: 202110752745.3
• 申請日: 2021-07-02
• 公開（公告）號: CN113382245A
• 公開（公告）日: 2021-09-10
• 發明（設計）人: 劉東;馮傲林;李禮
• 申請（專利權）人: 中國科學技術大學
• 主分類號: H04N19/147
• 分類號: H04N19/147;H04N19/149;H04N19/96;G06N3/04;G06N3/08
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 劉翠香

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中劃分深度存在異常的異常基本單元，并對所述異常基本單元的劃分深度進行修正。