[發明專利]圖像修復方法、裝置、終端及存儲介質

有效

• 申請號: 202010129488.3
• 申請日: 2020-02-28
• 公開（公告）號: CN111353965B
• 公開（公告）日: 2023-08-01
• 發明（設計）人: 朱圣晨
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50;G06T7/11;G06T7/136
• 代理公司: 北京三高永信知識產權代理有限責任公司 11138
• 代理人: 邢惠童

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

根據目標圖像和圖像分割模型，獲取第一掩膜圖像，所述圖像分割模型用于從所述目標圖像中分割出目標對象，所述目標對象是所述目標圖像中包含的任意一種物體對象；所述第一掩膜圖像是將所述目標圖像中所述目標對象的區域的像素值改為255，將其他區域的像素值改為0后得到的圖像；

確定用戶通過選擇工具在所述第一掩膜圖像或者所述目標圖像中確定的修復對象，所述修復對象是所述目標對象中的任意一個對象；

根據目標對象、所述修復對象與所述第一掩膜圖像，獲取第二掩膜圖像；所述第二掩膜圖像是保持所述第一掩膜圖像中的所述修復對象的像素值不變，將所述第一掩膜圖像中除所述修復對象之外的其他目標對象的像素值從255改為0后得到的圖像；

根據所述第二掩膜圖像和所述目標圖像，獲取待修復圖像；

根據所述待修復圖像和圖像修復模型，對所述待修復圖像進行修復，所述圖像修復模型是根據第一訓練數據訓練獲得的模型，所述第一訓練數據包括根據拍攝到的包含目標對象的圖像生成的受損圖像和拍攝到的不包含目標對象的圖像。

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

獲取第一場景下包含所述目標對象的圖像和所述第一場景下不包含所述目標對象的圖像；

根據所述第一場景下包含所述目標對象的圖像以及所述圖像分割模型，獲取第三掩膜圖像；

根據所述第三掩膜圖像和所述第一場景下包含所述目標對象的圖像，生成所述受損圖像；

根據所述受損圖像和所述第一場景下不包含目標對象產生的圖像，構建所述第一訓練數據。

3.根據權利要求1所述的方法，其特征在于，所述圖像修復模型中包含編碼網絡和解碼網絡；

所述根據所述待修復圖像和圖像修復模型，對所述待修復圖像進行修復，包括：

通過所述編碼網絡，提取所述待修復圖像的目標特征；

將所述目標特征輸入至所述解碼網絡中，獲取修復圖像。

4.根據權利要求3所述的方法，其特征在于，所述編碼網絡包含至少一個卷積層和至少一個空洞卷積層；

所述解碼網絡包含至少一個卷積層、至少一個空洞卷積層和至少一個上采樣層。

5.根據權利要求3所述的方法，其特征在于，所述圖像修復模型中還包含判別網絡，所述判別網絡用于計算所述圖像修復模型的損失函數。

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

響應于在目標界面中對圖像修復控件的選擇操作，獲取所述目標圖像。

7.一種圖像修復裝置，其特征在于，所述裝置包括：

第一獲取模塊，用于根據目標圖像和圖像分割模型，獲取第一掩膜圖像，所述圖像分割模型用于從所述目標圖像中分割出目標對象，所述目標對象是所述目標圖像中包含的任意一種物體對象；所述第一掩膜圖像是將所述目標圖像中所述目標對象的區域的像素值改為255，將其他區域的像素值改為0后得到的圖像；

對象確定模塊，用于確定用戶通過選擇工具在所述第一掩膜圖像或者所述目標圖像中確定的修復對象，所述修復對象是所述目標對象中的任意一個對象；

第二獲取模塊，用于根據目標對象、所述修復對象與所述第一掩膜圖像，獲取第二掩膜圖像；所述第二掩膜圖像是保持所述第一掩膜圖像中的所述修復對象的像素值不變，將所述第一掩膜圖像中除所述修復對象之外的其他目標對象的像素值從255改為0后得到的圖像；

待修復圖像獲取模塊，用于根據所述第二掩膜圖像和所述目標圖像，獲取待修復圖像；

圖像修復模塊，用于根據所述待修復圖像和圖像修復模型，對所述待修復圖像進行修復，所述圖像修復模型是根據第一訓練數據訓練獲得的模型，所述第一訓練數據包括根據拍攝到的包含目標對象的圖像生成的受損圖像和拍攝到的不包含目標對象的圖像。