[發明專利]圖像噪聲程度估計方法、裝置、電子設備及存儲介質

在審

• 申請號: 202110880750.2
• 申請日: 2021-08-02
• 公開（公告）號: CN113658118A
• 公開（公告）日: 2021-11-16
• 發明（設計）人: 吳飛
• 申請（專利權）人: 維沃移動通信有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T5/00;G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京東方億思知識產權代理有限責任公司 11258
• 代理人: 彭瓊

1.一種圖像噪聲程度估計方法，其特征在于，包括：

獲取第一圖像；

從所述第一圖像中提取與所述第一圖像對應的特征信息；

根據所述特征信息，確定與所述第一圖像對應的第一估計值和第二估計值；其中，所述第一估計值為所述第一圖像屬于噪聲圖像的估計值，所述第二估計值為所述第一圖像屬于無噪聲圖像的估計值；

計算所述第一估計值和所述第二估計值之間的差值；

將所述差值映射為預設連續區間范圍內的目標值，將所述目標值作為所述第一圖像的噪聲程度估計值。

2.根據權利要求1所述的方法，其特征在于，所述從所述第一圖像中提取與所述第一圖像對應的特征信息，包括：

將所述第一圖像轉換為對應的第二圖像；其中，所述第二圖像為拜耳Bayer圖像；

利用噪聲程度估計模型中的第一分支網絡，從所述第一圖像中提取與所述第一圖像對應的第一特征，以及，利用所述噪聲程度估計模型中的第二分支網絡，從所述第二圖像中提取與所述第一圖像對應的第二特征；

根據所述第一特征和所述第二特征，生成與所述第一圖像對應的特征信息。

3.根據權利要求1所述的方法，其特征在于，所述根據所述特征信息，確定與所述第一圖像對應的第一估計值和第二估計值，包括：

將所述特征信息輸入至噪聲程度估計模型中的二分類網絡，利用所述二分類網絡對所述第一圖像進行分類，輸出得到與所述第一圖像對應的第一估計值和第二估計值。

4.根據權利要求2所述的方法，其特征在于，在利用噪聲程度估計模型中的第一分支網絡，從所述第一圖像中提取與所述第一圖像對應的第一特征，以及，利用所述噪聲程度估計模型中的第二分支網絡，從所述第二圖像中提取與所述第一圖像對應的第二特征之前，所述方法還包括：

對所述第一圖像進行變換處理，得到與所述第一圖像對應的N個第三圖像；其中，所述變換處理包括變亮處理和變暗處理中的至少一項，N為正整數；

將所述N個第三圖像轉換為對應的N個第四圖像；其中，所述第四圖像為Bayer圖像；

所述利用噪聲程度估計模型中的第一分支網絡，從所述第一圖像中提取與所述第一圖像對應的第一特征，以及，利用所述噪聲程度估計模型中的第二分支網絡，從所述第二圖像中提取與所述第一圖像對應的第二特征，包括：

將所述第一圖像和所述N個第三圖像按通道維度堆疊，得到第一堆疊圖像，將所述第二圖像和所述N個第四圖像按通道維度堆疊，得到第二堆疊圖像；

利用所述第一分支網絡從所述第一堆疊圖像中提取顏色信息，得到與所述第一圖像對應的第一特征，利用所述第二分支網絡從所述第二堆疊圖像中提取空間信息，得到與所述第一圖像對應的第二特征。

5.根據權利要求4所述的方法，其特征在于，所述第一分支網絡包括第一特征提取子網絡和第一交叉注意力子網絡，所述第二分支網絡包括第二特征提取子網絡和第二交叉注意力子網絡；

所述利用所述第一分支網絡從所述第一堆疊圖像中提取顏色信息，得到與所述第一圖像對應的第一特征，利用所述第二分支網絡從所述第二堆疊圖像中提取空間信息，得到與所述第一圖像對應的第二特征，包括：

利用所述第一特征提取子網絡從所述第一堆疊圖像中提取顏色信息，得到與所述第一圖像對應的顏色特征，利用所述第二特征提取子網絡從所述第二堆疊圖像中提取空間信息，得到與所述第一圖像對應的空間特征；

利用所述第一交叉注意力子網絡，對所述顏色特征進行預處理，得到第一處理結果，利用所述第二交叉注意力子網絡，對所述空間特征進行預處理，得到第二處理結果；

將所述第一處理結果與所述空間特征進行融合，得到所述第二特征，將所述第二處理結果與所述顏色特征進行融合，得到所述第一特征。