[發明專利]圖像增強方法、智能設備、計算機存儲介質

有效

• 申請號: 202110755171.5
• 申請日: 2021-07-05
• 公開（公告）號: CN113436107B
• 公開（公告）日: 2023-06-20
• 發明（設計）人: 劉日升;石東祥;仲維;樊鑫;羅鐘鉉
• 申請（專利權）人: 鵬城實驗室;大連理工大學
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T7/90;G06V10/56;G06V10/82;G06N3/0464;G06N3/08
• 代理公司: 深圳市世紀恒程知識產權代理事務所 44287
• 代理人: 陳小娟

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

獲取水下初始圖像，并對所述水下初始圖像進行顏色空間轉化；

將所述水下初始圖像的L通道數據輸入對比度增強網絡中，通過對比度增強網絡中的特征提取模塊得到對應的圖像特征，并根據所述L通道數據生成自適應的濾波器，其中，所述對比度增強網絡由x個自適應卷積塊組成，每個自適應卷積塊使用y層全連接層網絡生成一個自適應的濾波器；

通過將所述圖像特征與生成的所述自適應的濾波器中的全連接層進行卷積，得到增強后的圖像特征；

將所述增強后的圖像特征輸入重建模塊，以通過所述重建模塊將所述增強后的圖像特征重建為第一圖像處理數據；

通過顏色矯正網絡對轉化后的所述水下初始圖像的AB通道數據進行處理，得到第二圖像處理數據，其中，所述L通道用于刻畫所述水下初始圖像的亮度，所述AB通道用于刻畫水下初始圖像的顏色；

根據所述第一圖像處理數據和第二圖像處理數據生成增強圖像。

2.如權利要求1所述的圖像增強方法，其特征在于，所述獲取水下初始圖像，并對所述水下初始圖像進行顏色空間轉化的步驟包括：

獲取水下初始圖像；

將所述水下初始圖像從RGB顏色空間轉化為LAB顏色空間。

3.如權利要求2所述的圖像增強方法，其特征在于，所述將所述水下初始圖像從RGB顏色空間轉化為LAB顏色空間的步驟包括：

將所述水下初始圖像從RGB顏色空間轉化為XYZ顏色空間；

將所述XYZ顏色空間轉化為LAB顏色空間。

4.如權利要求1所述的圖像增強方法，其特征在于，所述通過顏色矯正網絡對轉化后的所述水下初始圖像的AB通道數據進行處理，得到第二圖像處理數據的步驟包括：

將所述水下初始圖像的AB通道數據輸入顏色矯正網絡對應的編碼單元，其中，所述編碼單元中的下采樣層對所述AB通道數據進行下采樣，得到所述編碼單元的輸出圖像數據；

對所述輸出圖像數據進行特征提取，得到對應的顏色特征；

將所述顏色特征輸入所述顏色矯正網絡對應的解碼單元，其中，所述解碼單元中的上采樣塊對所述顏色特征進行上采樣，得到第二圖像處理數據。

5.如權利要求1所述的圖像增強方法，其特征在于，所述根據所述第一圖像處理數據和第二圖像處理數據生成增強圖像的步驟包括：

融合所述第一圖像處理數據與所述第二圖像處理數據，以得到初始增強圖像；

將所述初始增強圖像從LAB顏色空間轉化為RGB顏色空間，以生成所述增強圖像。

6.如權利要求1所述的圖像增強方法，其特征在于，所述將所述水下初始圖像的L通道數據輸入對比度增強網絡中，通過對比度增強網絡中的特征提取模塊得到對應的圖像特征，并根據所述L通道數據生成自適應的濾波器的步驟之前，還包括：

基于水上清晰圖像和深度圖像訓練所述對比度增強網絡。

7.如權利要求6所述的圖像增強方法，其特征在于，所述基于水上清晰圖像和深度圖像訓練所述對比度增強網絡的步驟包括：

獲取多組水上清晰圖像以及深度圖像；

根據所述深度圖像以及透射率參數，生成透射率圖；

根據所述水上清晰圖像以及所述透射率圖，生成低對比度圖像；

采用所述低對比度圖像訓練所述對比度增強網絡。

8.一種智能設備，其特征在于，所述智能設備包括存儲器、處理器以及存儲在所述存儲器并可在所述處理器上運行的圖像增強程序，所述圖像增強程序被所述處理器執行時實現如權利要求1-7任一項所述的圖像增強方法的各個步驟。

9.一種計算機存儲介質，其特征在于，所述計算機存儲介質存儲有圖像增強程序，所述圖像增強程序被處理器執行時實現如權利要求1-7任一項所述的圖像增強方法的各個步驟。