[發明專利]一種圖像修復方法、裝置及電子設備

在審

• 申請號: 202210514088.3
• 申請日: 2022-05-12
• 公開（公告）號: CN114841886A
• 公開（公告）日: 2022-08-02
• 發明（設計）人: 陶英杰
• 申請（專利權）人: 中國銀行股份有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50;G06V20/70
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 丁曼曼

本申請公開了一種圖像修復方法、裝置及電子設備，可應用于人工智能領域或金融領域。獲取待修復的目標圖像以及所述目標圖像的掩碼信息，對所述目標圖像以及所述掩碼信息進行部分卷積操作，得到初始圖像特征圖和初始掩碼信息，獲取粗糙圖像修復網絡模型，調用所述粗糙圖像修復網絡模型，得到粗糙修復特征圖和粗糙修復掩碼信息，獲取細節圖像修復網絡模型，連續多次調用所述細節圖像修復網絡模型，得到所述目標圖像對應的修復圖像。本發明中，通過對大區域缺失的圖像先進行粗修，得到粗糙的修復結果，然后通過精修的方式豐富缺失區域的紋理細節，實現了對大區域缺失的圖像的修復，滿足圖像修復需求。

技術領域

本發明涉及圖像修復領域，更具體的說，涉及一種圖像修復方法、裝置及電子設備。

背景技術

隨著科學技術的發展，圖像處理技術也日新月異，為人們的工作生活提供越來越多的便利。而在圖像的傳播或保存的過程中，往往存在受損，失真或被污染等造成信息不完整的問題。

為了避免信息不完整的問題，可以采用圖像修復方法進行修復。但是目前的圖像修復方法中，僅能夠修復小區域缺失的圖像，無法修復大區域缺失的圖像，進而無法滿足圖像修復需求。

發明內容

有鑒于此，本發明提供一種圖像修復方法、裝置及電子設備，以解決無法修復大區域缺失的圖像，進而無法滿足圖像修復需求的問題。

為解決上述技術問題，本發明采用了如下技術方案：

一種圖像修復方法，包括：

獲取待修復的目標圖像以及所述目標圖像的掩碼信息；

對所述目標圖像以及所述掩碼信息進行部分卷積操作，得到初始圖像特征圖和初始掩碼信息；

獲取具有不同卷積屬性的多個多尺度修復模塊構建得到的粗糙圖像修復網絡模型；所述多尺度修復模塊由兩條具有不同卷積屬性的并行卷積支路構成；

調用所述粗糙圖像修復網絡模型，對所述初始圖像特征圖和初始掩碼信息分別進行處理，得到所述初始圖像特征圖對應的中間圖像特征圖和所述初始掩碼信息對應的中間掩碼信息，并對所述中間圖像特征圖和所述中間掩碼信息進行遞歸計算，得到粗糙修復特征圖和粗糙修復掩碼信息；

獲取用于對圖像進行細節修復的細節圖像修復網絡模型；

連續多次調用所述細節圖像修復網絡模型對所述粗糙修復特征圖、所述粗糙修復掩碼信息、所述初始圖像特征圖和所述初始掩碼信息進行處理，得到所述目標圖像的多個細節修復結果，并對所述多個細節修復結果進行合并處理，得到所述目標圖像對應的修復圖像。

可選地，調用所述粗糙圖像修復網絡模型，對所述初始圖像特征圖和初始掩碼信息分別進行處理，得到所述初始圖像特征圖對應的中間圖像特征圖和所述初始掩碼信息對應的中間掩碼信息，并對所述中間圖像特征圖和所述中間掩碼信息進行遞歸計算，得到粗糙修復特征圖和粗糙修復掩碼信息，包括：

調用所述粗糙圖像修復網絡模型，以使所述粗糙圖像修復網絡模型中的多尺度修復模塊的并行卷積支路，分別對所述初始圖像特征圖進行處理，得到第一圖像特征圖和第二圖像特征圖，以及分別對所述初始掩碼信息進行處理，得到第一掩碼和第二掩碼；

基于所述第一掩碼和所述第二掩碼，計算掩碼結果，并將所述掩碼結果的平均值，作為所述初始掩碼信息對應的中間掩碼信息；

基于所述掩碼結果、所述第一圖像特征圖和所述第二圖像特征圖，計算圖像處理結果，并將所述圖像處理結果的平均值，作為所述初始圖像特征圖對應的中間圖像特征圖；

對所述中間圖像特征圖和所述中間掩碼信息進行遞歸計算，得到粗糙修復特征圖和粗糙修復掩碼信息。