[發明專利]圖像修復方法、裝置、計算機設備及存儲介質

有效

• 申請號: 202010130428.3
• 申請日: 2020-02-28
• 公開（公告）號: CN111353956B
• 公開（公告）日: 2022-11-15
• 發明（設計）人: 朱圣晨
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50;G06T3/40;G06V10/52
• 代理公司: 北京三高永信知識產權代理有限責任公司 11138
• 代理人: 邢惠童

1.一種圖像修復方法，其特征在于，所述方法包括：

獲取待修復圖像；

根據所述待修復圖像，獲取掩膜圖像；所述掩膜圖像與所述待修復圖像的分辨率大小一致；

根據所述待修復圖像，獲取所述待修復圖像的參數矩陣；根據所述掩膜圖像，獲取所述掩膜圖像的參數矩陣；將所述待修復圖像的參數矩陣和所述掩膜圖像的參數矩陣進行拼接，獲取目標圖像矩陣；所述目標圖像矩陣包含所述待修復圖像的各個像素點的參數，以及所述待修復圖像的掩膜圖像的各個像素點的參數；所述參數包括各個像素點的像素值、色溫值、亮度值中的至少一個；

通過圖像修復模型的至少兩個分支結構，獲取所述目標圖像矩陣的目標特征；所述至少兩個分支結構用于提取目標圖像矩陣的不同特征尺度的特征；

根據所述目標特征獲取修復圖像。

2.根據權利要求1所述的方法，其特征在于，所述通過圖像修復模型的至少兩個分支結構，獲取所述目標圖像矩陣的目標特征，包括：

通過第一分支結構，提取所述目標圖像矩陣的第一目標特征，所述第一分支結構是所述至少兩個分支結構中的任意一個分支結構；

將各個分支結構得到的各自的第一目標特征進行拼接，獲取所述目標特征。

3.根據權利要求2所述的方法，其特征在于，所述第一分支結構中包含至少一個卷積層模塊、至少一個空洞卷積層模塊以及至少一個上采樣模塊；

所述通過第一分支結構，提取所述目標圖像矩陣的第一目標特征，包括：

通過所述第一分支結構中的所述至少一個卷積層模塊、所述至少一個空洞卷積層模塊以及所述至少一個上采樣模塊，對所述目標圖像矩陣進行處理獲取所述第一目標特征。

4.根據權利要求3所述的方法，其特征在于，所述第一分支結構中包含的各個卷積核大小均相同；并且，

不同的分支結構中各自包含的卷積核大小不同。

5.根據權利要求1至4所述的方法，其特征在于，所述方法還包括：

對目標數據集中的圖像進行剪裁，獲取至少兩個固定大小的原始圖像；

利用各個所述原始圖像訓練所述圖像修復模型。

6.根據權利要求5所述的方法，其特征在于，所述利用各個所述原始圖像訓練所述圖像修復模型，包括：

根據第一原始圖像，獲取第一受損圖像，所述第一原始圖像是所述至少兩個固定大小的原始圖像中的任意一個原始圖像；

對所述第一受損圖像生成第一修復圖像；

根據所述第一修復圖像以及所述第一原始圖像計算損失函數；

根據所述損失函數，對所述圖像修復模型進行迭代訓練，直至所述圖像修復模型收斂。

7.根據權利要求6所述的方法，其特征在于，所述根據所述第一修復圖像以及所述第一原始圖像計算損失函數，包括：

根據第一判別網絡計算第一歐式距離，所述第一歐式距離是所述第一修復圖像以及所述第一原始圖像之間的歐式距離；

根據第二判別網絡計算第二歐式距離，所述第二歐式距離是所述第一修復圖像的受損區域與所述第一原始圖像的受損區域之間的歐式距離；

將所述第一歐式距離與所述第二歐式距離的和作為所述損失函數。