[發明專利]圖像增強方法、裝置及電子設備

有效

• 申請號: 202011036003.2
• 申請日: 2020-09-27
• 公開（公告）號: CN112102204B
• 公開（公告）日: 2022-07-08
• 發明（設計）人: 王詩韻;李瑮;吳家新
• 申請（專利權）人: 蘇州科達科技股份有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T7/90
• 代理公司: 北京三聚陽光知識產權代理有限公司 11250
• 代理人: 張琳琳

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

獲取原始圖像對應的待處理圖像并提取所述原始圖像的色彩信息以及飽和度信息；

將所述待處理圖像輸入至亮度增強模型中，得到增強后的亮度信息；

對所述色彩信息、所述飽和度信息以及所述增強后的亮度信息進行融合，確定增強后的所述待處理圖像；

其中，所述將所述待處理圖像輸入至亮度增強模型中，得到增強后的亮度信息，包括：

利用所述亮度增強模型中的至少一個卷積單元對所述待處理圖像進行至少一次的特征提取；

利用所述亮度增強模型中的至少一個注意力機制單元對所述卷積單元或轉置卷積單元的輸出信息進行圖像細節的恢復；

利用所述亮度增強模型中的至少一個轉置卷積單元對所述注意力機制單元的輸出信息進行解碼；

將與所述亮度增強模型的輸入層連接的第一個卷積單元的輸出信息與最后一個所述注意力機制單元的輸出信息進行融合，確定所述增強后的亮度信息。

2.根據權利要求1所述的方法，其特征在于，所述利用所述亮度增強模型中的至少一個注意力機制單元對所述卷積單元或轉置卷積單元的輸出信息進行圖像細節的恢復，包括：

利用所述注意力機制單元中的卷積層對所述卷積單元的輸出信息，或所述轉置卷積單元的輸出信息進行卷積處理，得到第一輸出信息；

對所述第一輸出信息進行全局池化以及全連接處理，得到第二輸出信息；

計算所述第一輸出信息與所述第二輸出信息的乘積，得到對應于所述卷積單元的輸出信息的圖像細節恢復結果或對應于所述轉置卷積單元的輸出信息的圖像細節恢復結果。

3.根據權利要求1所述的方法，其特征在于，所述將所述待處理圖像輸入至亮度增強模型中，得到增強后的亮度信息，包括：

將所述待處理圖像輸入所述亮度增強模型的第一個卷積單元，得到第一個卷積單元的特征提取結果；

將所述第一個卷積單元的特征提取結果輸入第一卷積單元中，得到第一卷積單元的特征提取結果，所述第一卷積單元包括至少一個卷積單元；

將第一卷積單元的特征提取結果以及第一轉置卷積單元的解碼結果輸入所述第一注意力機制單元中，得到圖像細節恢復的結果；其中，所述第一轉置卷積單元的輸入為第二注意力機制單元的輸出；

將所述第一注意力機制單元的輸出與所述第一個卷積單元的輸出進行融合，確定增強后的亮度信息。

4.根據權利要求1所述的方法，其特征在于，所述將與所述亮度增強模型的輸入層連接的第一個卷積單元的輸出信息與最后一個所述注意力機制單元的輸出信息進行融合，確定所述增強后的亮度信息，包括：

計算所述第一個卷積單元的輸出信息與最后一個所述注意力機制單元的輸出信息之和，得到第三輸出信息；

對所述第三輸出信息進行卷積處理，得到所述增強后的亮度信息。

5.根據權利要求1所述的方法，其特征在于，所述獲取原始圖像對應的待處理圖像，包括：

獲取原始圖像；

對所述原始圖像進行預處理得到所述待處理圖像，所述待處理圖像中的各個像素點按預設規律排列。

6.根據權利要求1-5中任一項所述的方法，其特征在于，所述亮度增強模型是通過如下方式訓練得到的：

獲取至少一個場景的兩種亮度信息的樣本圖像，得到與第一亮度信息對應的第一樣本圖像，以及與第二亮度信息對應的第二樣本圖像；

將所述第一樣本圖像輸入亮度增強模型中，得到增強后的亮度信息；

基于增強后的亮度信息與對應的所述第二樣本圖像的亮度信息，對所述亮度增強模型的參數進行更新。

7.根據權利要求6所述的方法，其特征在于，所述獲取至少一個場景的兩種亮度信息的樣本圖像，得到與第一亮度信息對應的第一樣本圖像，以及與第二亮度信息對應的第二樣本圖像，包括：

獲取至少一個場景的兩種亮度信息的原始樣本圖像；

對所述原始樣本圖像進行預處理，得到所述第一樣本圖像以及所述第二樣本圖像，所述第一樣本圖像以及所述第二樣本圖像中的各個像素點按預設規律排列。