[發(fā)明專利]圖像降噪處理方法、裝置、電子設(shè)備及存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202110791758.1
• 申請日: 2021-07-13
• 公開（公告）號: CN113362259A
• 公開（公告）日: 2021-09-07
• 發(fā)明（設(shè)計）人: 張祎;李大松;秦紅偉;王曉剛;李鴻升
• 申請（專利權(quán)）人: 商湯集團有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00
• 代理公司: 廣州三環(huán)專利商標(biāo)代理有限公司 44202
• 代理人: 熊永強;董文俊

1.一種圖像降噪處理方法，其特征在于，包括：

獲取待處理噪聲圖像；

基于目標(biāo)降噪網(wǎng)絡(luò)對所述待處理噪聲圖像進行降噪處理，得到目標(biāo)降噪圖像；

其中，所述目標(biāo)降噪網(wǎng)絡(luò)為基于初始噪聲圖像集進行多次降噪迭代操作得到的標(biāo)簽圖像集對待訓(xùn)練降噪網(wǎng)絡(luò)進行降噪訓(xùn)練得到的。

2.根據(jù)權(quán)利要求1所述的圖像降噪處理方法，其特征在于，在所述獲取待處理噪聲圖像之前，所述方法還包括：

基于所述初始噪聲圖像集進行N次降噪迭代操作，得到所述標(biāo)簽圖像集，其中，第i次降噪迭代操作包括：將第(i-1)次降噪迭代操作得到的降噪圖像集作為所述第i次降噪迭代操作的標(biāo)簽圖像集；對所述第i次降噪迭代操作的標(biāo)簽圖像集進行加噪處理，得到第i次降噪迭代操作的加噪圖像集；基于所述第i次降噪迭代操作的標(biāo)簽圖像集和所述第i次降噪迭代操作的加噪圖像集對待訓(xùn)練降噪網(wǎng)絡(luò)進行降噪訓(xùn)練，得到所述第i次降噪迭代操作的待優(yōu)化降噪網(wǎng)絡(luò)；基于所述第i次降噪迭代操作的待優(yōu)化降噪網(wǎng)絡(luò)對所述初始噪聲圖像集進行降噪處理，得到第i次降噪迭代操作的降噪圖像集；

將第N次降噪迭代操作過程中得到的待優(yōu)化降噪網(wǎng)絡(luò)，作為所述目標(biāo)降噪網(wǎng)絡(luò)；

其中，N為大于1的正整數(shù)，i為大于等于1且小于等于N的正整數(shù)；當(dāng)i＝1時，所述第i-1次降噪迭代操作得到的降噪圖像集為所述初始噪聲圖像集。

3.根據(jù)權(quán)利要求2所述的圖像降噪處理方法，其特征在于，所述基于所述第i次降噪迭代操作的標(biāo)簽圖像集和所述第i次降噪迭代操作的加噪圖像集對待訓(xùn)練降噪網(wǎng)絡(luò)進行降噪訓(xùn)練，得到所述第i次降噪迭代操作的待優(yōu)化降噪網(wǎng)絡(luò)包括：

將所述第i次降噪迭代操作的加噪圖像集輸入所述待訓(xùn)練降噪網(wǎng)絡(luò)進行降噪處理，得到第i次降噪迭代操作的預(yù)測降噪圖像集；

根據(jù)所述第i次降噪迭代操作的標(biāo)簽圖像集和所述第i次降噪迭代操作的預(yù)測降噪圖像集確定第i次降噪迭代操作的降噪損失信息；

基于所述第i次降噪迭代操作的降噪損失信息更新所述待訓(xùn)練降噪網(wǎng)絡(luò)的網(wǎng)絡(luò)參數(shù)；

基于更新網(wǎng)絡(luò)參數(shù)后的待訓(xùn)練降噪網(wǎng)絡(luò)，重復(fù)所述將所述第i次降噪迭代操作的加噪圖像集輸入所述待訓(xùn)練降噪網(wǎng)絡(luò)進行降噪處理，得到第i次降噪迭代操作的預(yù)測降噪圖像集至所述基于所述第i次降噪迭代操作的降噪損失信息更新所述待訓(xùn)練降噪網(wǎng)絡(luò)的網(wǎng)絡(luò)參數(shù)的訓(xùn)練迭代操作，至達到預(yù)設(shè)訓(xùn)練收斂條件；

將達到所述預(yù)設(shè)訓(xùn)練收斂條件的情況下得到的待訓(xùn)練降噪網(wǎng)絡(luò)，作為所述第i次降噪迭代操作的待優(yōu)化降噪網(wǎng)絡(luò)。

4.根據(jù)權(quán)利要求2所述的圖像降噪處理方法，其特征在于，所述方法還包括：

確定目標(biāo)噪聲類型對應(yīng)的預(yù)設(shè)噪聲強度區(qū)間；

所述對所述第i次降噪迭代操作的標(biāo)簽圖像集進行加噪處理，得到第i次降噪迭代操作的加噪圖像集包括：

根據(jù)所述預(yù)設(shè)噪聲強度區(qū)間，隨機生成與所述第i次降噪迭代操作的標(biāo)簽圖像集中圖像對應(yīng)的目標(biāo)強度的噪聲數(shù)據(jù)；

基于所述噪聲數(shù)據(jù)對所述第i次降噪迭代操作的標(biāo)簽圖像集進行加噪處理，得到所述第i次降噪迭代操作的加噪圖像集。

5.根據(jù)權(quán)利要求3所述的圖像降噪處理方法，其特征在于，所述N次降噪迭代操作中任意一次降噪迭代操作對應(yīng)的待訓(xùn)練降噪網(wǎng)絡(luò)為對預(yù)設(shè)神經(jīng)網(wǎng)絡(luò)進行隨機初始化得到的網(wǎng)絡(luò)。

6.根據(jù)權(quán)利要求5所述的圖像降噪處理方法，其特征在于，所述預(yù)設(shè)訓(xùn)練收斂條件包括訓(xùn)練迭代操作的次數(shù)達到第一預(yù)設(shè)訓(xùn)練次數(shù)。

7.根據(jù)權(quán)利要求3所述的圖像降噪處理方法，其特征在于，所述N次降噪迭代操作中第(i+1)次降噪迭代操作對應(yīng)的待訓(xùn)練降噪網(wǎng)絡(luò)為第i次降噪迭代操作得到的待優(yōu)化降噪網(wǎng)絡(luò)。