[發(fā)明專利]圖像修復(fù)方法、裝置、終端及存儲(chǔ)介質(zhì)

有效

• 申請(qǐng)?zhí)枺?/td>: 202010129488.3
• 申請(qǐng)日: 2020-02-28
• 公開（公告）號(hào): CN111353965B
• 公開（公告）日: 2023-08-01
• 發(fā)明（設(shè)計(jì)）人: 朱圣晨
• 申請(qǐng)（專利權(quán)）人: OPPO廣東移動(dòng)通信有限公司
• 主分類號(hào): G06T5/50
• 分類號(hào): G06T5/50;G06T7/11;G06T7/136
• 代理公司: 北京三高永信知識(shí)產(chǎn)權(quán)代理有限責(zé)任公司 11138
• 代理人: 邢惠童

本申請(qǐng)公開了一種圖像修復(fù)方法、裝置、終端及存儲(chǔ)介質(zhì)，屬于圖像處理技術(shù)領(lǐng)域。所述方法包括：根據(jù)目標(biāo)圖像和圖像分割模型，獲取第一掩膜圖像；根據(jù)目標(biāo)對(duì)象與第一掩膜圖像，獲取第二掩膜圖像，根據(jù)第二掩膜圖像和目標(biāo)圖像，獲取待修復(fù)圖像；根據(jù)待修復(fù)圖像和圖像修復(fù)模型，對(duì)待修復(fù)圖像進(jìn)行修復(fù)，圖像修復(fù)模型是根據(jù)第一訓(xùn)練數(shù)據(jù)訓(xùn)練獲得的模型，第一訓(xùn)練數(shù)據(jù)包括根據(jù)拍攝到的包含目標(biāo)對(duì)象的圖像生成的受損圖像和拍攝到的不包含目標(biāo)對(duì)象的圖像。本申請(qǐng)通過從第二掩膜圖像和目標(biāo)圖像獲取待修復(fù)圖像，并對(duì)該待修復(fù)圖像進(jìn)行修復(fù)，使得用戶可以得到在目標(biāo)圖像中去除任一目標(biāo)對(duì)象后的修復(fù)圖像，提高了得到修復(fù)圖像的效率。

技術(shù)領(lǐng)域

本申請(qǐng)實(shí)施例涉及圖像處理技術(shù)領(lǐng)域，特別涉及一種圖像修復(fù)方法、裝置、終端及存儲(chǔ)介質(zhì)。

背景技術(shù)

隨著圖像處理技術(shù)的快速發(fā)展，在人們的日常生活中，各種各樣的終端中已經(jīng)具有拍攝并處理圖像的功能。

其中，用戶可以對(duì)終端中存儲(chǔ)的圖像進(jìn)行處理，也可以對(duì)終端當(dāng)前拍攝的圖像進(jìn)行處理。比如，當(dāng)終端中的某個(gè)圖像出現(xiàn)破損時(shí)，終端可以將該圖像修復(fù)。當(dāng)用戶拍攝到圖片后，終端可以對(duì)該圖像進(jìn)行調(diào)色、剪切等處理。目前，對(duì)于終端中的某個(gè)圖像，如果需要去除這個(gè)圖像中包含的某個(gè)對(duì)象，并保證去除后的區(qū)域與該圖像中的場(chǎng)景符合，通常需要使用多種圖像處理軟件對(duì)該圖像進(jìn)行處理。

在相關(guān)技術(shù)中，采用多種圖像處理軟件對(duì)圖像進(jìn)行處理的過程繁瑣，出錯(cuò)率高，導(dǎo)致獲取用戶需要的圖像的效率低。

發(fā)明內(nèi)容

本申請(qǐng)實(shí)施例提供了一種圖像修復(fù)方法、裝置、終端及存儲(chǔ)介質(zhì)，提高了獲取修復(fù)圖像的效率。所述技術(shù)方案如下：

一方面，本申請(qǐng)實(shí)施例提供了一種圖像修復(fù)方法，所述方法包括：

根據(jù)目標(biāo)圖像和圖像分割模型，獲取第一掩膜圖像，所述圖像分割模型用于從所述目標(biāo)圖像中分割出目標(biāo)對(duì)象，所述目標(biāo)對(duì)象是所述目標(biāo)圖像中包含的任意一個(gè)物體對(duì)象；

根據(jù)目標(biāo)對(duì)象與所述第一掩膜圖像，獲取第二掩膜圖像；

根據(jù)所述第二掩膜圖像和所述目標(biāo)圖像，獲取待修復(fù)圖像；

根據(jù)所述待修復(fù)圖像和圖像修復(fù)模型，對(duì)所述待修復(fù)圖像進(jìn)行修復(fù)，所述圖像修復(fù)模型是根據(jù)第一訓(xùn)練數(shù)據(jù)訓(xùn)練獲得的模型，所述第一訓(xùn)練數(shù)據(jù)包括根據(jù)拍攝到的包含目標(biāo)對(duì)象的圖像生成的受損圖像和拍攝到的不包含目標(biāo)對(duì)象的圖像。

另一方面，本申請(qǐng)實(shí)施例提供了一種圖像修復(fù)裝置，所述裝置包括：

第一獲取模塊，用于根據(jù)目標(biāo)圖像和圖像分割模型，獲取第一掩膜圖像，所述圖像分割模型用于從所述目標(biāo)圖像中分割出目標(biāo)對(duì)象，所述目標(biāo)對(duì)象是所述目標(biāo)圖像中包含的任意一個(gè)物體對(duì)象；

第二獲取模塊，用于根據(jù)目標(biāo)對(duì)象與所述第一掩膜圖像，獲取第二掩膜圖像；

待修復(fù)圖像獲取模塊，用于根據(jù)所述第二掩膜圖像和所述目標(biāo)圖像，獲取待修復(fù)圖像；

圖像修復(fù)模塊，用于根據(jù)所述待修復(fù)圖像和圖像修復(fù)模型，對(duì)所述待修復(fù)圖像進(jìn)行修復(fù)，所述圖像修復(fù)模型是根據(jù)第一訓(xùn)練數(shù)據(jù)訓(xùn)練獲得的模型，所述第一訓(xùn)練數(shù)據(jù)包括根據(jù)拍攝到的包含目標(biāo)對(duì)象的圖像生成的受損圖像和拍攝到的不包含目標(biāo)對(duì)象的圖像。

另一方面，本申請(qǐng)實(shí)施例提供了一種終端，所述終端包含處理器和存儲(chǔ)器，所述存儲(chǔ)器中存儲(chǔ)有至少一條指令、至少一段程序、代碼集或指令集，所述至少一條指令、所述至少一段程序、所述代碼集或指令集由所述處理器加載并執(zhí)行以實(shí)現(xiàn)如上述一個(gè)方面所述的圖像修復(fù)方法。

另一方面，本申請(qǐng)實(shí)施例提供了一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，所述存儲(chǔ)介質(zhì)中存儲(chǔ)有至少一條指令、至少一段程序、代碼集或指令集，所述至少一條指令、所述至少一段程序、所述代碼集或指令集由處理器加載并執(zhí)行以實(shí)現(xiàn)如上述一個(gè)方面的圖像修復(fù)方法。