[發明專利]一種圖像增強方法及計算設備

有效

• 申請號: 201811252617.7
• 申請日: 2018-10-25
• 公開（公告）號: CN109345487B
• 公開（公告）日: 2020-12-25
• 發明（設計）人: 周銘柯;李志陽;張偉;李啟東;許清泉
• 申請（專利權）人: 廈門美圖之家科技有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50;G06T7/90;G06N3/08;G06N3/04
• 代理公司: 北京思睿峰知識產權代理有限公司 11396
• 代理人: 高攀;趙愛軍

1.一種圖像增強方法，所述方法在計算設備中執行，所述方法包括：

將待處理圖像輸入預設的圖像增強模型中，經多次卷積處理后輸出圖像；

將所述待處理圖像和輸出圖像分別轉換到預定顏色空間，其中，所述預定顏色空間為Lab顏色空間；以及

在預定顏色空間中融合待處理圖像和輸出圖像生成增強圖像；

其中，所述將待處理圖像和輸出圖像分別轉換到預定顏色空間的步驟還包括：

將待處理圖像轉換到預定顏色空間，得到三個通道上的第一待處理圖、第二待處理圖和第三待處理圖；

將輸出圖像轉換到預定顏色空間，得到三個通道上的第一輸出圖、第二輸出圖和第三輸出圖；

其中，所述在預定顏色空間中融合待處理圖像和輸出圖像生成增強圖像的步驟包括：

判斷第一待處理圖中像素點的像素值大小、并根據判斷結果結合第一輸出圖生成第一增強圖；

結合第二待處理圖和第二輸出圖生成第二增強圖；

結合第三待處理圖和第三輸出圖生成第三增強圖；以及

融合所述第一增強圖、第二增強圖和第三增強圖生成增強圖像。

2.如權利要求1所述的方法，其中，根據判斷結果結合第一輸出圖生成第一增強圖的步驟包括：

若第一待處理圖中像素點的像素值小于第一閾值，則將第一輸出圖對應像素點的像素值作為第一增強圖對應像素點的像素值；

若第一待處理圖中像素點的像素值大于第二閾值，則結合第一待處理圖和第一輸出圖中對應像素點的像素值以第一方式生成第一增強圖對應像素點的像素值；以及

若第一待處理圖中像素點的像素值既不小于第一閾值也不大于第二閾值，則結合第一待處理圖和第一輸出圖中對應像素點的像素值以第二方式生成第一增強圖對應像素點的像素值。

3.如權利要求1或2所述的方法，其中，所述結合第二待處理圖和第二輸出圖生成第二增強圖的步驟包括：

將第二待處理圖和第二輸出圖進行加權計算，生成第二增強圖。

4.如權利要求3所述的方法，其中，所述結合第三待處理圖和第三輸出圖生成第三增強圖的步驟包括：

將第三待處理圖和第三輸出圖進行加權計算，生成第三增強圖。

5.如權利要求1所述的方法，其中，所述圖像增強模型包括依次相連的多個中間處理塊和一個結果處理塊，其中，

每個所述中間處理塊包括依次相連的至少兩個卷積激活層和一個跳躍連接層，所述跳躍連接層適于將其所屬的中間處理塊的第一個卷積激活層的輸入與最后一個卷積激活層的輸出相加；

所述結果處理塊包括多個卷積激活層。

6.如權利要求5所述的方法，其中，在所述圖像增強模型的第一個中間處理塊之前，還包括一個卷積激活層。

7.如權利要求6所述的方法，其中，各中間處理塊中的卷積激活層的激活函數為ReLU函數。

8.如權利要求7所述的方法，其中，所述中間處理塊的個數為4。

9.如權利要求8所述的方法，其中，所述結果處理塊包括三個卷積激活層，其中，前兩個卷積激活層的激活函數為ReLU函數，第三個卷積激活層的激活函數為Tanh函數。

10.如權利要求1所述的方法，在將待處理圖像輸入預設的圖像增強模型的步驟之前，還包括通過訓練生成預設的圖像增強模型的步驟：

獲取多個訓練圖像對，每一個訓練圖像對包括輸入圖像和目標圖像，其中所述輸入圖像為通過單反相機采集的圖像，所述目標圖像為對輸入圖像進行調整處理后的圖像；

將所述輸入圖像輸入到預訓練的圖像增強模型，經多次卷積處理后輸出圖像，按照預設損失函數計算輸出圖像相對于目標圖像的損失值并更新圖像增強模型的參數，直到所述損失值滿足預定條件時，訓練結束，得到訓練后的圖像增強模型，作為預設的圖像增強模型。