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

有效

• 申請?zhí)枺?/td>: 202110655903.3
• 申請日: 2021-06-11
• 公開（公告）號: CN113382243B
• 公開（公告）日: 2022-11-01
• 發(fā)明（設計）人: 不公告發(fā)明人
• 申請（專利權）人: 上海壁仞智能科技有限公司
• 主分類號: H04N19/146
• 分類號: H04N19/146;H04N19/30;H04N19/48
• 代理公司: 北京路浩知識產(chǎn)權代理有限公司 11002
• 代理人: 程琛

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

確定待壓縮圖像，并給定一個初始圖像生成網(wǎng)絡和一個低行列數(shù)的多通道二維隨機數(shù)組；所述隨機數(shù)組的行列數(shù)小于所述待壓縮圖像的分辨率；

基于所述待壓縮圖像以及所述隨機數(shù)組，對所述初始圖像生成網(wǎng)絡進行迭代訓練，得到圖像生成網(wǎng)絡，并確定所述圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)；所述網(wǎng)絡參數(shù)的數(shù)量小于所述待壓縮圖像的像素個數(shù)；

將所述圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)以及所述隨機數(shù)組作為所述待壓縮圖像的壓縮信息表示。

2.根據(jù)權利要求1所述的圖像壓縮方法，其特征在于，還包括：

以分辨率相同的多個子圖像對所述待壓縮圖像進行分割，所述多個子圖像的分辨率給定，所述多個子圖像所合成的區(qū)域大于或等于所述待壓縮圖像；其中，當所述多個子圖像所合成的區(qū)域大于所述待壓縮圖像時，對所述多個子圖像所合成的區(qū)域中未包含所述待壓縮圖像的部分進行隨機填充。

3.根據(jù)權利要求2所述的圖像壓縮方法，其特征在于，所述基于所述待壓縮圖像以及所述隨機數(shù)組，對所述初始圖像生成網(wǎng)絡進行迭代訓練，包括：

將所述多個子圖像中每一個子圖像所對應的隨機數(shù)組輸入至所述初始圖像生成網(wǎng)絡，得到所述初始圖像生成網(wǎng)絡輸出的所述隨機數(shù)組對應的生成子圖像；

基于每一個所述子圖像以及所述對應的生成子圖像，對所述初始圖像生成網(wǎng)絡進行迭代訓練，以得到每一個所述子圖像所對應的圖像生成網(wǎng)絡，并確定所述圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)。

4.根據(jù)權利要求3所述的圖像壓縮方法，其特征在于，所述將所述多個子圖像中每一個子圖像所對應的隨機數(shù)組輸入至所述初始圖像生成網(wǎng)絡，得到所述初始圖像生成網(wǎng)絡輸出的所述隨機數(shù)組對應的生成子圖像，包括：

將每一個所述子圖像所對應的隨機數(shù)組輸入至所述初始圖像生成網(wǎng)絡所包含的至少一個上采樣層中的第一個上采樣層，由所述至少一個上采樣層對所述隨機數(shù)組進行逐級上采樣，依次得到至少一個更高行列數(shù)的多通道二維采樣隨機數(shù)組，所述至少一個采樣隨機數(shù)組中的最后一個采樣隨機數(shù)組的行列數(shù)與對應子圖像的分辨率相同；

將所述最后一個采樣隨機數(shù)組輸入至所述初始圖像生成網(wǎng)絡的圖像轉換層，得到所述圖像轉換層輸出的所述隨機數(shù)組對應的生成子圖像。

5.根據(jù)權利要求4所述的圖像壓縮方法，其特征在于，所述初始圖像生成網(wǎng)絡還包括位于每一個所述上采樣層之后的自定義網(wǎng)絡層。

6.根據(jù)權利要求3所述的圖像壓縮方法，其特征在于，在確定所述圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)，之后還包括：

以所述圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)更新所述初始圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)后，將所述隨機數(shù)組輸入至所述初始圖像生成網(wǎng)絡，得到所述初始圖像生成網(wǎng)絡輸出的解壓子圖像；

對各解壓子圖像進行合并，得到所述待壓縮圖像。

7.根據(jù)權利要求1至6任一項所述的圖像壓縮方法，其特征在于，所述初始圖像生成網(wǎng)絡的網(wǎng)絡參數(shù)包括固定網(wǎng)絡參數(shù)和訓練網(wǎng)絡參數(shù)，所述固定網(wǎng)絡參數(shù)是預先設置的；

所述對所述初始圖像生成網(wǎng)絡進行迭代訓練，包括：

僅對所述初始圖像生成網(wǎng)絡的所述訓練網(wǎng)絡參數(shù)進行迭代訓練。

8.根據(jù)權利要求1至6任一項所述的圖像壓縮方法，其特征在于，所述隨機數(shù)組包括多個子數(shù)組；

所述基于所述待壓縮圖像以及所述隨機數(shù)組，對所述初始圖像生成網(wǎng)絡進行迭代訓練，得到圖像生成網(wǎng)絡，包括：

基于所述待壓縮圖像以及所述隨機數(shù)組中的各子數(shù)組，分別對所述初始圖像生成網(wǎng)絡進行迭代訓練，得到各子數(shù)組對應的圖像生成子網(wǎng)絡，并確定各圖像生成子網(wǎng)絡的損失函數(shù)值；

將迭代后最小損失函數(shù)值對應的圖像生成子網(wǎng)絡作為所述圖像生成網(wǎng)絡。