[發明專利]HEVC幀內角度模式選擇方法、裝置、設備及可讀存儲介質

有效

• 申請號: 202010628727.X
• 申請日: 2020-07-02
• 公開（公告）號: CN111800642B
• 公開（公告）日: 2023-05-26
• 發明（設計）人: 李帥廠;陳小磊;劉俊;姜光心
• 申請（專利權）人: 中實燃氣發展(西安)有限公司
• 主分類號: H04N19/593
• 分類號: H04N19/593;H04N19/105;G06N3/0464;G06N3/08
• 代理公司: 西安通大專利代理有限責任公司 61200
• 代理人: 朱海臨

1.一種HEVC幀內角度模式選擇方法，其特征在于，包括以下步驟：

S1：將若干輸入視頻幀圖像分別劃分為若干PU圖像，將所有PU圖像分類為訓練集和測試集；將訓練集內所有PU圖像相同位置像素值計算均值得到均值圖像；

S2：構建深度卷積神經網絡，并將深度卷積神經網絡的輸出設定為35類角度模式；通過訓練集和均值圖像對深度卷積神經網絡進行訓練，通過測試集對深度卷積神經網絡進行測試，得到深度卷積神經網絡預測模型；

深度卷積神經網絡包括依次設置的第一卷積層、第一池化層、第二卷積層、第二池化層、第一全連接層、第二全連接層以及輸出層，輸出層的輸出設定為35類角度模式；第一卷積層和第二卷積層均設定為64個卷積核，卷積核大小均為5×5；第一池化層和第二池化層均設定為MaxPooling操作，池化窗口大小均為3×3；第一全連接層和第二全連接層均設定為512維神經元節點；

S3：獲取待預測視頻幀圖像，并將其劃分為若干待預測PU圖像，將若干待預測PU圖像通過深度卷積神經網絡預測模型進行預測，得到每個待預測PU圖像對應的角度模式；

具體的，按照HEVC原始的四叉樹劃分方式獲取待預測視頻幀圖像所有的需要的PU圖像，并對PU圖像進行統一尺寸處理，統一為8×8，然后對每個像素點減去均值，得到待預測PU圖像；

S4：選取數量最多的前預設個數類角度模式組成角度模式候選列表，然后通過HEVC編碼流程遍歷角度模式候選列表中的角度模式，得到最優角度模式。

2.根據權利要求1所述的HEVC幀內角度模式選擇方法，其特征在于，所述S1的具體方法為：

將若干輸入視頻幀圖像分別按照4種PU圖像尺寸進行無重疊劃分，將4種尺寸的PU圖像全部縮放為同一尺寸，然后將同一尺寸的PU圖像分類為訓練集和測試集。

3.根據權利要求1所述的HEVC幀內角度模式選擇方法，其特征在于，所述S1中，將輸入視頻幀圖像的前50幀的PU圖像作為訓練集，其余幀的PU圖像作為測試集。

4.根據權利要求1所述的HEVC幀內角度模式選擇方法，其特征在于，所述S2中通過訓練集和均值圖像對深度卷積神經網絡進行訓練時，以及通過測試集對深度卷積神經網絡進行測試時，均通過最小二乘法函數和反向傳播BP算法對深度卷積神經網絡進行更新與求解。

5.根據權利要求1所述的HEVC幀內角度模式選擇方法，其特征在于，所述S4中，選取數量最多的前3個角度模式組成角度模式候選列表。

6.根據權利要求5所述的HEVC幀內角度模式選擇方法，其特征在于，所述S4中，當數量最多的前3類角度模式中不包含Planar模式和/或DC模式時，將Planar模式和/或DC模式添加至角度模式候選列表。