[發明專利]網絡訓練方法和圖像處理方法、裝置、設備以及存儲介質

在審

• 申請號: 202210926739.X
• 申請日: 2022-08-03
• 公開（公告）號: CN115187846A
• 公開（公告）日: 2022-10-14
• 發明（設計）人: 張順意;白國財;王超鋒;駱遙;種道晨
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V10/82
• 分類號: G06V10/82;G06V10/774;G06N3/08;G06N3/04
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 丁蕓;馬敬

本公開提供了一種網絡訓練方法和圖像處理方法、裝置、設備以及存儲介質，涉及圖像處理技術領域，尤其涉及深度學習、人工智能、計算機視覺等領域。具體實現方案為：獲取多個樣本圖像，以及各個樣本圖像對應的蒙版圖像；獲取多個真實圖像，真實圖像為不包含目標對象的圖像；針對每一樣本圖像，將樣本圖像和樣本圖像對應的蒙版圖像輸入生成對抗網絡的生成模型，通過生成模型輸出生成圖像；將生成圖像和真實圖像輸入判別模型，通過判別模型得到針對生成圖像和真實圖像的輸出；基于輸出調整模型參數，得到訓練后的生成對抗網絡，生成對抗網絡包括訓練后的生成模型。本公開能夠實現利用生成對抗網絡的生成模型對圖像中目標對象的抹除。

技術領域

本公開涉及圖像處理技術領域，尤其涉及深度學習、人工智能、計算機視覺等領域。

背景技術

圖像處理過程中，有對圖像中某些信息進行抹除的需求，例如，需要抹除圖像的車輛、人等物體。

發明內容

本公開提供了一種圖像處理方法、裝置、設備以及存儲介質。

根據本公開的第一方面，提供了一種生成對抗網絡訓練方法，所述生成對抗網絡包括生成模型和判別模型，包括：

獲取多個樣本圖像，以及各個樣本圖像對應的蒙版圖像；

獲取多個真實圖像，所述真實圖像為不包含目標對象的圖像；

針對每一樣本圖像，將所述樣本圖像和所述樣本圖像對應的蒙版圖像輸入所述生成對抗網絡的生成模型，通過所述生成模型輸出生成圖像；

將所述生成圖像和所述真實圖像輸入所述判別模型，通過所述判別模型得到針對所述生成圖像和所述真實圖像的輸出；

基于所述輸出調整模型參數，得到訓練后的生成對抗網絡，所述生成對抗網絡包括訓練后的生成模型。

根據本公開的第二方面，提供了一種圖像處理方法，包括：

獲取待處理圖像；

對所述待處理圖像進行圖像識別，得到所述待處理圖像中的目標對象；

確定針對所述目標對象的蒙版圖像；

將所述待處理圖像和所述蒙版圖像輸入生成模型，通過所述生成模型輸出抹除后圖像，所述抹除后圖像表示從所述待處理圖像中抹除所述目標對象后的圖像，其中，所述生成模型是生成對抗網絡中的生成模型，所述生成對抗網絡通過上述第一方面所述的生成對抗網絡訓練方法訓練得到。

根據本公開的第三方面，提供了一種生成對抗網絡訓練裝置，所述生成對抗網絡包括生成模型和判別模型，所述裝置包括：

獲取模塊，用于獲取多個樣本圖像，以及各個樣本圖像對應的蒙版圖像；獲取多個真實圖像，所述真實圖像為不包含目標對象的圖像；

生成模塊，用于針對每一樣本圖像，將所述樣本圖像和所述樣本圖像對應的蒙版圖像輸入所述生成模型，通過所述生成模型輸出生成圖像；

判別模塊，用于將所述生成圖像和所述真實圖像輸入所述判別模型，通過所述判別模型得到針對所述生成圖像和所述真實圖像的輸出；

訓練模塊，用于基于所述輸出調整模型參數，得到訓練后的生成對抗網絡，所述生成對抗網絡包括訓練后的生成模型。

根據本公開的第四方面，提供了一種圖像處理裝置，包括：

獲取模塊，用于獲取待處理圖像；

識別模塊，用于對所述待處理圖像進行圖像識別，得到所述待處理圖像中的目標對象；

確定模塊，用于確定針對所述目標對象的蒙版圖像；