[發明專利]視頻編碼方法和裝置

在審

• 申請號: 202210969495.3
• 申請日: 2022-08-12
• 公開（公告）號: CN115484464A
• 公開（公告）日: 2022-12-16
• 發明（設計）人: 楊明田;張昊;王豪;陳高星;繆澤翔;蘇昊天;石雋
• 申請（專利權）人: 阿里巴巴（中國）有限公司
• 主分類號: H04N19/96
• 分類號: H04N19/96;H04N19/436;H04N19/30;H04N19/119;H04N19/103
• 代理公司: 北京君以信知識產權代理有限公司 11789
• 代理人: 郝玉娥

1.一種視頻編碼方法，其特征在于，包括：

獲取視頻幀的編碼樹單元；

獲取所述編碼樹單元的預測殘差圖像和預測重建圖像；

通過多任務學習的預測模型，根據所述預測殘差圖像和預測重建圖像，獲取所述編碼樹單元的分層編碼單元劃分圖和分層編碼單元模式圖，所述分層編碼單元劃分圖包括編碼單元的劃分概率，所述分層編碼單元模式圖包括編碼單元的編碼模式概率；

根據所述劃分概率、所述編碼模式概率，確定所述編碼樹單元的劃分模式及編碼模式。

2.根據權利要求1所述的方法，其特征在于，所述通過多任務學習的預測模型，根據所述預測殘差圖像和預測重建圖像，獲取所述編碼樹單元的分層編碼單元劃分圖和分層編碼單元模式圖，包括：

通過所述預測模型，根據所述預測殘差圖像和預測重建圖像，獲取共享特征；

根據所述共享特征，獲取所述分層編碼單元劃分圖；

根據所述共享特征，獲取所述分層編碼單元模式圖。

3.根據權利要求2所述的方法，其特征在于，

所述通過所述預測模型，根據所述預測殘差圖像和預測重建圖像，獲取共享特征，包括：

通過所述預測模型，根據所述預測殘差圖像和預測重建圖像，獲取多個層次的共享特征；

所述根據所述共享特征，獲取所述分層編碼單元劃分圖，包括：

根據多個層次的共享特征，獲取所述分層編碼單元劃分圖；

所述根據所述共享特征，獲取所述分層編碼單元模式圖，包括：

根據多個層次的共享特征，獲取所述分層編碼單元模式圖。

4.根據權利要求3所述的方法，其特征在于，

所述根據多個層次的共享特征，獲取所述分層編碼單元劃分圖，包括：

根據第一層次、第二層次和第三層次分別對應的共享特征，獲取所述分層編碼單元劃分圖；

所述根據多個層次的共享特征，獲取所述分層編碼單元模式圖，包括：

根據第一層次、第二層次、第三層次和第四層次分別對應的共享特征，獲取所述分層編碼單元模式圖。

5.根據權利要求1-4任一項所述的方法，其特征在于，所述獲取所述編碼樹單元的預測殘差圖像和預測重建圖像，包括：

將對所述編碼樹單元進行幀間編碼產生的重建圖像作為所述預測重建圖像；

根據預測重建圖像確定所述預測殘差圖像。

6.根據權利要求1-4任一項所述的方法，其特征在于，所述根據所述劃分概率、所述編碼模式概率，確定所述編碼樹單元的劃分模式及編碼模式，包括：

根據所述劃分概率，判斷是否跳過編碼單元在當前深度的編碼過程和/或跳過編碼單元劃分模式編碼過程；

根據編碼模式概率，判斷是否跳過編碼單元的幀內編碼過程或者幀間編碼過程。

7.一種視頻編碼預測模型處理方法，其特征在于，包括：

獲取訓練數據集；訓練數據包括：殘差圖像和重建圖像與分層編碼單元劃分圖的標注數據和分層編碼單元模式圖的標注數據之間的對應關系；

構建多任務學習的預測模型；

根據所述訓練數據集，訓練所述預測模型的模型參數。

8.一種視頻編碼裝置，其特征在于，包括：

編碼樹單元獲取單元，用于獲取視頻幀的編碼樹單元；

第一編碼單元，用于獲取所述編碼樹單元的預測殘差圖像和預測重建圖像；

預測單元，用于通過多任務學習的預測模型，根據所述預測殘差圖像和預測重建圖像，獲取所述編碼樹單元的分層編碼單元劃分圖和分層編碼單元模式圖，所述分層編碼單元劃分圖包括編碼單元的劃分概率，所述分層編碼單元模式圖包括編碼單元的編碼模式概率；

第二編碼單元，用于根據所述劃分概率、所述編碼模式概率，確定所述編碼樹單元的劃分模式及編碼模式。

9.一種電子設備，其特征在于，包括：

處理器和存儲器；

存儲器，用于存儲實現根據權利要求1-7任一項所述的方法的程序，該設備通電并通過所述處理器運行該方法的程序。