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

在審

• 申請號: 202111164844.6
• 申請日: 2021-09-30
• 公開（公告）號: CN113888431A
• 公開（公告）日: 2022-01-04
• 發明（設計）人: 宮振飛
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/50;G06T7/13;G06K9/62
• 代理公司: 深圳市智圈知識產權代理事務所(普通合伙) 44351
• 代理人: 呂靜

本申請公開了一種圖像修復模型的訓練方法、裝置、計算機設備及存儲介質，該圖像修復模型的訓練方法包括：獲取樣本圖像對應的掩膜圖像；提取樣本圖像的邊緣圖像作為第一邊緣圖像；基于預先訓練的邊緣補全模型，對第一邊緣圖像進行邊緣補全，得到補全后的邊緣圖像作為第二邊緣圖像；基于掩膜圖像，獲取樣本圖像對應的修復掩膜圖像以及上下文掩膜圖像，修復掩膜圖像用于指示樣本圖像中待修復的區域，上下文掩膜圖像用于指示對待修復的區域進行修復時的已知圖像區域；基于樣本圖像、修復掩膜圖像、上下文掩膜圖像以及第二邊緣圖像，對初始模型進行訓練，得到訓練后的圖像修復模型。本方法可以實現提升圖像修復模型的圖像修復效果。

技術領域

本申請涉及圖像處理技術領域，更具體地，涉及一種圖像修復模型的訓練方法、裝置、計算機設備及存儲介質。

背景技術

隨著科技水平和生活水平的快速進步，電子設備(例如智能手機、平板電腦等)被人們廣泛使用，而且電子設備的拍攝功能也越來越強，非專業人員也能通過電子設備拍攝出較高質量的圖像。但是在圖像的拍攝、傳輸、處理以及保存過程中，由于各種因素，可能導致圖像存在缺失、裂痕等。因此，出現了對這些缺陷圖像進行修復的方法，但目前的圖像修復方法的圖像修復效果不佳。

發明內容

鑒于上述問題，本申請提出了一種圖像修復模型的訓練方法、裝置、計算機設備及存儲介質。

第一方面，本申請實施例提供了一種圖像修復模型的訓練方法，所述方法包括：獲取樣本圖像對應的掩膜圖像，所述掩膜圖像用于指示所述樣本圖像中的顯著性物體所在區域；提取所述樣本圖像的邊緣圖像作為第一邊緣圖像，所述邊緣圖像用于表征所述樣本圖像的圖像內容的紋理；基于預先訓練的邊緣補全模型，對所述第一邊緣圖像進行邊緣補全，得到補全后的邊緣圖像作為第二邊緣圖像；基于所述樣本圖像、所述掩膜圖像以及所述第二邊緣圖像，對初始模型進行訓練，得到訓練后的圖像修復模型。

第二方面，本申請實施例提供了一種圖像修復模型的訓練裝置，所述裝置包括：掩膜獲取模塊、第一邊緣獲取模塊、第二邊緣獲取模塊以及模型訓練模塊，其中，所述掩膜獲取模塊用于獲取樣本圖像對應的掩膜圖像，所述掩膜圖像用于指示所述樣本圖像中的顯著性物體所在區域；所述第一邊緣獲取模塊用于提取所述樣本圖像的邊緣圖像作為第一邊緣圖像，所述邊緣圖像用于表征所述樣本圖像的圖像內容的紋理；所述第二邊緣獲取模塊用于基于預先訓練的邊緣補全模型，對所述第一邊緣圖像進行邊緣補全，得到補全后的邊緣圖像作為第二邊緣圖像；所述模型訓練模塊用于基于所述樣本圖像、所述掩膜圖像以及所述第二邊緣圖像，對初始模型進行訓練，得到訓練后的圖像修復模型。

第三方面，本申請實施例提供了一種計算機設備，包括：一個或多個處理器；存儲器；一個或多個應用程序，其中所述一個或多個應用程序被存儲在所述存儲器中并被配置為由所述一個或多個處理器執行，所述一個或多個程序配置用于執行上述第一方面提供的圖像修復模型的訓練方法。

第四方面，本申請實施例提供了一種計算機可讀取存儲介質，所述計算機可讀取存儲介質中存儲有程序代碼，所述程序代碼可被處理器調用執行上述第一方面提供的圖像修復模型的訓練方法。

本申請提供的方案，通過獲取樣本圖像對應的掩膜圖像，掩膜圖像用于指示樣本圖像中的顯著性物體所在區域，提取樣本圖像的邊緣圖像作為第一邊緣圖像，邊緣圖像用于表征樣本圖像的圖像內容的紋理，基于預先訓練的邊緣補全模型，對第一邊緣圖像進行邊緣補全，得到補全后的邊緣圖像作為第二邊緣圖像，基于樣本圖像、掩膜圖像以及第二邊緣圖像，對初始模型進行訓練，得到訓練后的圖像修復模型。由于在模型訓練時，對提取的邊緣圖像進行了邊緣補全，并且基于掩膜圖像精細化了待修復的區域以及已知圖像區域，因此訓練得到的模型能夠更好地修復圖像內容，提升圖像修復效果。

附圖說明

為了更清楚地說明本申請實施例中的技術方案，下面將對實施例描述中所需要使用的附圖作簡單地介紹，顯而易見地，下面描述中的附圖僅僅是本申請的一些實施例，對于本領域技術人員來講，在不付出創造性勞動的前提下，還可以根據這些附圖獲得其他的附圖。