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

有效

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

本發明涉及圖像處理技術領域，具體設備及圖像增強方法、裝置及電子設備，所述方法包括獲取原始圖像對應的待處理圖像并提取原始圖像的色彩信息以及飽和度信息；將所述待處理圖像輸入至亮度增強模型中，得到增強后的亮度信息；對所述色彩信息、所述飽和度信息以及所述增強后的亮度信息進行融合，確定增強后的所述待處理圖像。將待處理圖像作為亮度增強模型的輸入，輸出為增強后的亮度信息，且將增強后的亮度信息與原始圖像的色彩信息以及飽和度信息進行融合，所得到的增強后的待處理圖像能保持待處理圖像的色彩信息以及飽和度信息不變，僅對亮度信息進行增強，從而有效解決了增強后圖像的偏色問題。

技術領域

本發明涉及圖像處理技術領域，具體涉及圖像增強方法、裝置及電子設備。

背景技術

圖像在較低的光照環境下或者背光的條件下往往會存在亮度低、噪點多以及對比差等問題，從主觀上會影響到人們對圖片的感受，同時也會影響后續在圖像像素級別上的相關任務，如夜晚視頻監控下的車牌識別、背光條件下的人臉識別等。通過圖像增強技術能夠減少圖像中的噪聲，提高物體與背景之間的對比度，改善圖像的視覺效果同時也是后續相關識別任務的基礎。因此，低照圖像增強方法的研究具有比較重要的現實意義和價值。

目前所采用的圖像增強方法是獲取到圖像傳感器采集到的圖像，將圖像輸入至圖像增強模型中，輸出為對輸入圖像的整體增強。然而，這一處理方式中，由于在對圖像增強的過程中是對圖像的整體進行增強的，就會導致增強處理后的圖像容易出現偏色問題。

發明內容

有鑒于此，本發明實施例提供了一種圖像增強方法、裝置及電子設備，以解決增強后圖像的偏色問題。

根據第一方面，本發明實施例提供了一種圖像增強方法，包括：

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

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

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

本發明實施例提供的圖像增強方法，將待處理圖像作為亮度增強模型的輸入，輸出為增強后的亮度信息，且將增強后的亮度信息與原始圖像的色彩信息以及飽和度信息進行融合，所得到的增強后的待處理圖像能保持待處理圖像的色彩信息以及飽和度信息不變，僅對亮度信息進行增強，從而有效解決了增強后圖像的偏色問題。進一步地，由于亮度增強模型是僅對亮度信息進行增強，能夠減少內存的開銷，使得該圖像增強方法能夠應用在終端設備上，對終端設備采集到的圖像進行實時的圖像增強。

結合第一方面，在第一方面第一實施方式中，所述將所述待處理圖像輸入至亮度增強模型中，得到增強后的亮度信息，包括：

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

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

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

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

本發明實施例提供的圖像增強方法，在亮度增強模型中利用轉置卷積單元實現對注意力機制單元的輸出信息的解碼，利用注意力機制單元實現圖像細節的恢復，從而該亮度增強模型能夠更加準確地實現對待處理圖像的亮度信息進行增強。

結合第一方面第一實施方式，在第一方面第二實施方式中，所述利用所述亮度增強模型中的至少一個注意力機制單元對所述卷積單元或轉置卷積單元的輸出信息進行圖像細節的恢復，包括：