[發明專利]模型生成方法、圖像增強方法、裝置及計算機可讀存儲介質

有效

• 申請號: 201811594319.6
• 申請日: 2018-12-25
• 公開（公告）號: CN109859152B
• 公開（公告）日: 2021-05-25
• 發明（設計）人: 鐘韜
• 申請（專利權）人: 北京奇藝世紀科技有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50
• 代理公司: 北京潤澤恒知識產權代理有限公司 11319
• 代理人: 莎日娜

1.一種模型生成方法，其特征在于，所述方法包括：

對于不同強度等級差對應的樣本圖像對，根據所述樣本圖像對中的樣本圖像及所述樣本圖像對的強度等級差，并利用初始增強模型對所述樣本圖像進行增強處理；所述強度等級差表示所述樣本圖像對中樣本增強圖像的強度等級與所述樣本圖像的強度等級之間的差值；

基于增強后的樣本圖像以及所述樣本圖像的樣本增強圖像，獲取所述初始增強模型的損失值；

若所述損失值在預設范圍內，將所述初始增強模型作為圖像增強模型；所述圖像增強模型接收輸入的圖像以及目標強度等級差，并輸出目標圖像；

所述初始增強模型包括卷積層、全連接層以及反卷積層；

所述卷積層中的參數至少包括：卷積核的值，所述全連接層中的參數至少包括權重矩陣；

所述根據所述樣本圖像對中的樣本圖像及所述樣本圖像對的強度等級差，并利用所述初始增強模型對所述樣本圖像進行增強處理，包括：

將所述樣本圖像對及所述樣本圖像對的強度等級差輸入初始增強模型；

通過所述卷積層將所述樣本圖像中每個像素組成的像素矩陣，與所述卷積核進行卷積運算，得到卷積后的第一像素矩陣；

通過所述全連接層將所述權重矩陣、所述第一像素矩陣以及所述樣本圖像對的強度等級差相乘，得到第二像素矩陣；

通過所述反卷積層對所述第二像素矩陣進行反卷積運算，得到增強后的樣本圖像。

2.根據權利要求1所述的方法，其特征在于，所述基于增強后的樣本圖像以及所述樣本圖像的樣本增強圖像，獲取所述初始增強模型的損失值之后，所述方法還包括：

若所述損失值不在預設范圍內，基于所述損失值表示的誤差程度調整所述初始增強模型的參數，并基于所述不同強度等級差對應的樣本圖像對，對調整后的初始增強模型繼續訓練。

3.根據權利要求1所述的方法，其特征在于，所述根據所述樣本圖像對中的樣本圖像及所述樣本圖像對的強度等級差，并利用所述初始增強模型對所述樣本圖像進行增強處理之前，所述方法還包括：

獲取多個初始圖像對；所述初始圖像對中包括初始圖像以及初始圖像的增強圖像；

對于每個初始圖像對，按照不同強度等級對應的圖像參數調整范圍以及不同強度等級對應的圖像種類，對所述初始圖像對中的初始圖像及增強圖像的圖像參數進行調整，得到對應不同強度等級的備選圖像；

利用每個備選圖像對應的強度等級，對所述備選圖像進行組合，得到不同強度等級差對應的樣本圖像對；所述圖像參數至少包括亮度、對比度及飽和度。

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

將待處理圖像及目標強度等級差輸入圖像增強模型；

根據所述目標強度等級差，并通過所述圖像增強模型對所述待處理圖像進行增強處理，得到目標圖像；其中，所述圖像增強模型是利用權利要求1至3中任一項所述的方法生成的；

所述圖像增強模型包括卷積層、全連接層以及反卷積層；

所述根據所述目標強度等級差，并通過所述圖像增強模型對所述待處理圖像進行增強處理，得到目標圖像，包括：

通過所述卷積層將所述待處理圖像中每個像素組成的像素矩陣，與所述卷積層中的卷積核進行卷積運算，得到卷積后的第一像素矩陣；

通過所述全連接層將所述全連接層中的權重矩陣、所述第一像素矩陣以及所述目標強度等級差相乘，得到第二像素矩陣；

通過所述反卷積層對所述第二像素矩陣進行反卷積運算，得到目標圖像。