[發明專利]圖像壓縮方法、裝置、電子設備和存儲介質

在審

• 申請號: 202210515370.3
• 申請日: 2022-05-11
• 公開（公告）號: CN114979665A
• 公開（公告）日: 2022-08-30
• 發明（設計）人: 張墨琪;鄧偉輝;李小成
• 申請（專利權）人: 北京字跳網絡技術有限公司
• 主分類號: H04N19/42
• 分類號: H04N19/42;H04N19/436
• 代理公司: 北京遠智匯知識產權代理有限公司 11659
• 代理人: 魯艷萍

1.一種圖像壓縮方法，其特征在于，包括：

獲取待壓縮圖像；

將所述待壓縮圖像輸入到預設圖像壓縮模型中，得到與所述待壓縮圖像對應的目標壓縮圖像；

其中，所述預設圖像壓縮模型的模型訓練樣本對包括原始樣本圖像和壓縮樣本圖像，其中，所述壓縮樣本圖像是采用預設圖像編解碼器對所述原始樣本圖像進行壓縮處理得到的圖像。

2.根據權利要求1所述的方法，其特征在于，所述預設圖像壓縮模型對所述待壓縮圖像進行圖像壓縮的過程，包括：

對所述待壓縮圖像進行圖像預處理，得到預設尺寸的一個或多個子待壓縮圖像塊；

分別將各所述子待壓縮圖像塊輸入到預設圖像壓縮神經網絡中進行圖像壓縮，得到對應的子圖像壓縮結果；

將各所述子圖像壓縮結果進行圖像拼接，得到所述目標壓縮圖像。

3.根據權利要求2所述的方法，其特征在于，所述預設圖像壓縮神經網絡的訓練過程，包括：

獲取所述模型訓練樣本對經過所述圖像預處理，得到的一個或多個原始樣本圖像塊和壓縮樣本圖像塊，并將各所述原始樣本圖像塊輸入到初始圖像壓縮網絡中的編碼器中，進行特征提取，得到對應的樣本編碼特征；

基于第一預設解碼器對所述樣本編碼特征進行特征解碼處理，得到初步壓縮圖像塊，并基于所述初步壓縮圖像塊和對應的所述壓縮樣本圖像塊計算第一損失函數；

基于第二預設解碼器對所述樣本編碼特征進行特征解碼處理，得到初步重建圖像塊，并基于所述初步重建圖像塊和所述原始樣本圖像塊或與所述原始樣本圖像塊對應的高分辨率原始圖像塊，計算第二損失函數；

根據所述第一損失函數和所述第二損失函數的結果迭代更新所述初始圖像壓縮網絡，當所述第一損失函數和所述第二損失函數的加權和值滿足預設收斂條件時，完成模型訓練過程。

4.根據權利要求2所述的方法，其特征在于，所述對所述待壓縮圖像進行圖像預處理，得到預設尺寸的多個子待壓縮圖像塊，包括：

對所述待壓縮圖像進行邊緣填充，得到目標尺寸圖像；

將所述目標尺寸圖像按照預設尺寸進行切割，得到預設尺寸的多個子待壓縮圖像塊。

5.根據權利要求1所述的方法，其特征在于，所述模型訓練樣本對的構建過程包括：

獲取多個原始樣本圖像，并將各所述原始樣本圖像分別輸入到所述預設圖像編解碼器，得到對應的配對壓縮圖像；

分別將各所述原始樣本圖像與對應的配對壓縮圖像作為一組樣本對，得到第一樣本集合。

6.根據權利要求5所述的方法，其特征在于，所述模型訓練樣本對的構建過程，還包括：

在得到所述配對壓縮圖像之后，對所述配對壓縮圖像進行圖像優化處理，得到壓縮后優化圖像；

分別將各所述原始樣本圖像與對應的壓縮后優化圖像作為一組樣本對，得到第二樣本集合。

7.根據權利要求6所述的方法，其特征在于，當以所述第二樣本集合中的樣本對作為所述預設圖像壓縮模型的訓練樣本時，所述預設圖像壓縮模型中的預設圖像壓縮神經網絡為設置有殘差學習機制的網絡。

8.根據權利要求1所述的方法，其特征在于，所述方法還包括：

將所述目標壓縮圖像作為所述待壓縮圖像的預壓縮圖像，輸入至所述預設圖像壓縮模型和/或與預設圖像壓縮格式對應的圖像編解碼器中進行圖像壓縮，得到最終壓縮圖像。

9.根據權利要求1-8中任一所述的方法，其特征在于，在將所述待壓縮圖像輸入到預設圖像壓縮模型之前，所述方法還包括：

當所述待壓縮圖像的第一像素信息表示格式和與所述預設圖像壓縮模型適配的第二像素信息表示格式不同時，將所述待壓縮圖像的像素信息表示格式轉換為所述第二像素信息表示格式；

相應的，在得到所述目標壓縮圖像之后，所述方法還包括：

將所述目標壓縮圖像的像素信息表示格式轉換為所述第一像素信息表示格式；

其中，所述第二像素信息表示格式為與所述預設圖像編解碼器適配的像素信息表示格式。