[發明專利]一種圖像處理方法、裝置、終端設備及存儲介質

在審

• 申請號: 202110882192.3
• 申請日: 2021-08-02
• 公開（公告）號: CN113781320A
• 公開（公告）日: 2021-12-10
• 發明（設計）人: 章政文;陳翔宇;董超;喬宇
• 申請（專利權）人: 中國科學院深圳先進技術研究院
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50
• 代理公司: 深圳中一聯合知識產權代理有限公司 44414
• 代理人: 胡明強

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

利用已訓練的優化模型對待處理圖像進行目標類型優化處理，得到初始優化圖像；

通過已訓練的局部增強模型對所述初始優化圖像進行局部增強處理，得到增強圖像；

將所述增強圖像和所述待處理圖像的過曝掩模圖像輸入到已訓練的補償模型中進行處理，對所述增強圖像的高亮區域進行信息補償，得到補償圖像，所述過曝掩模圖像指示所述高亮區域。

2.根據權利要求1所述的方法，其特征在于，所述局部增強模型包括：下采樣模塊、上采樣模塊以及設置在所述下采樣模塊和所述上采樣模塊之間的多個殘差網絡。

3.根據權利要求1所述的方法，其特征在于，所述過曝掩模圖像中像素點的像素值的確定方法包括：

根據公式確定所述過曝掩模圖像中像素點的像素值，其中，I_mask(x,y)表示所述過曝掩模圖像位于(x,y)處的像素點的像素值，I_s(x,y)表示所述待處理圖像位于(x,y)處的像素點的像素值，λ表示預設的過曝閾值。

4.根據權利要求1所述的方法，其特征在于，所述補償模型包括生成器；

所述將所述增強圖像和所述待處理圖像的過曝掩模圖像輸入到已訓練的補償模型中，對所述增強圖像的高亮區域進行信息補償，包括：

將所述增強圖像輸入到已訓練的所述生成器中進行處理，得到全局曝光信息；

根據所述待處理圖像的過曝掩模圖像和所述全局曝光信息確定所述高亮區域的過曝光信息；

利用所述過曝光信息對所述高亮區域進行補償，得到所述補償圖像。

5.根據權利要求1所述的方法，其特征在于，分別對優化初始模型、局部增強初始模型和補償初始模型進行訓練，得到對應的所述優化模型、所述局部增強模型和所述補償模型。

6.根據權利要求4所述的方法，其特征在于，所述生成器的訓練方法包括：

構建生成對抗網絡，所述生成對抗網絡包括所述生成器的初始模型和判別器；

利用預設的損失函數和訓練集對所述生成對抗網絡進行對抗訓練，得到所述生成器，其中，訓練集包括與多個待處理圖像樣本對應的增強圖像樣本、過曝掩模圖像樣本和補償圖像樣本；

所述損失函數用于描述所述補償圖像樣本與預測圖像之間的絕對誤差損失值、所述補償圖像樣本與所述預測圖像之間的感知損失值和所述預測圖像的判別器損失值的綜合損失值；所述預測圖像是指所述增強圖像樣本經過所述生成器的初始模型處理后，與所述過曝掩模圖像樣本相乘，再與所述增強圖像樣本疊加得到的圖像。

7.根據權利要求1至6任一項所述的方法，其特征在于，所述目標類型優化處理是指HDR轉換處理，所述待處理圖像為從SDR視頻中抽幀得到的視頻幀，所述SDR視頻中的每一幀視頻幀依次經過所述優化模型、所述局部增強模型和所述補償模型處理后輸出的所述補償圖像，經過合幀后得到與所述SDR視頻對應的HDR視頻。

8.一種圖像處理裝置，其特征在于，包括：

優化單元，用于利用已訓練的優化模型對待處理圖像進行目標類型優化處理，得到初始優化圖像；

增強單元，用于通過已訓練的局部增強模型對所述初始優化圖像進行局部增強處理，得到增強圖像；

補償單元，用于將所述增強圖像和所述待處理圖像的過曝掩模圖像輸入到已訓練的補償模型中進行處理，對所述增強圖像的高亮區域進行信息補償，得到補償圖像，所述過曝掩模圖像指示所述高亮區域。

9.一種終端設備，包括存儲器、處理器以及存儲在所述存儲器中并可在所述處理器上運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現如權利要求1至7任一項所述的方法。

10.一種計算機可讀存儲介質，所述計算機可讀存儲介質存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現如權利要求1至7任一項所述的方法。