[發明專利]圖像處理方法、裝置、電子設備和存儲介質

在審

• 申請號: 202011105958.9
• 申請日: 2020-10-15
• 公開（公告）號: CN112258404A
• 公開（公告）日: 2021-01-22
• 發明（設計）人: 郝燕龍;張淵;郭小燕;鄭文
• 申請（專利權）人: 北京達佳互聯信息技術有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 林祥

1.一種圖像處理方法，其特征在于，包括：

獲取待修復圖像和所述待修復圖像對應的掩碼圖像，所述待修復圖像包括待修復的目標區域和所述目標區域之外的背景區域，所述掩碼圖像用于指示所述目標區域和所述背景區域的相對位置關系；

將所述待修復圖像和所述掩碼圖像輸入預訓練的第一圖像修復網絡中，以使所述第一圖像修復網絡對所述目標區域進行修復處理得到第一過渡圖像，所述第一過渡圖像包括對應于所述目標區域的過渡目標區域；

按照所述相對位置關系，將所述待修復圖像的背景區域與所述第一過渡圖像的過渡目標區域進行合并得到第二過渡圖像；

將所述第二過渡圖像輸入預訓練的第二圖像修復網絡中，以使所述第二圖像修復網絡對所述過渡目標區域進行修復處理得到修復后圖像。

2.根據權利要求1所述的方法，其特征在于，所述掩碼圖像包括對應于所述目標區域的目標掩碼區域和對應于所述背景區域的背景掩碼區域，所述目標掩碼區域中像素點的顏色值與所述背景掩碼區域中像素點的顏色值不同。

3.根據權利要求1所述的方法，其特征在于，所述獲取所述待修復圖像對應的掩碼圖像，包括：

將所述待修復圖像輸入預訓練的對象分割網絡，以使所述對象分割網絡識別所述待修復圖像中的目標對象，并對所述目標對象所在的目標掩碼區域和所述目標掩碼區域之外的背景掩碼區域進行二值化處理得到掩碼圖像，二值化后的所述掩碼圖像中各個像素點的取值用于表征所述相對位置關系。

4.根據權利要求1所述的方法，其特征在于，將所述待修復圖像輸入所述第一圖像修復網絡，包括：

對所述待修復圖像進行歸一化處理得到歸一化待修復圖像，并將所述歸一化待修復圖像輸入所述圖像修復網絡。

5.根據權利要求1所述的方法，其特征在于，所述第一圖像修復網絡和所述第二圖像修復網絡的訓練方法，包括：

通過在樣本原始圖像中添加目標對象得到樣本待修復圖像，并確定包括所述樣本待修復圖像和對應于所述樣本待修復圖像的樣本掩碼圖像的訓練樣本，所述樣本待修復圖像包括對應于所述目標對象的待修復的樣本目標區域和所述樣本目標區域之外的樣本背景區域，所述樣本掩碼圖像用于指示所述樣本目標區域和所述樣本背景區域的相對位置關系；

將所述訓練樣本輸入第一待訓練網絡，以獲取所述第一待訓練網絡對所述樣本目標區域進行修復處理得到的第一樣本過渡圖像，所述第一樣本過渡圖像包括對應于所述樣本目標區域的樣本過渡目標區域；

按照所述相對位置關系，將所述待修復圖像的背景區域與所述第一樣本過渡圖像的樣本過渡目標區域進行合并，并將合并得到的第二樣本過渡圖像輸入第二待訓練網絡，以獲取所述第二待訓練網絡對所述樣本過渡目標區域進行修復處理得到的樣本修復后圖像，所述樣本修復后圖像包括對應于所述樣本目標區域的樣本修復后目標區域；

根據所述樣本原始圖像和所述樣本修復后圖像訓練所述第一待訓練網絡和所述第二待訓練網絡，以得到所述第一圖像修復網絡和所述第二圖像修復網絡。

6.根據權利要求1所述的方法，其特征在于，所述第一圖像修復網絡和所述第二圖像修復網絡基于U-Net結構的生成對抗網絡構建。

7.根據權利要求6所述的方法，其特征在于，所述第一圖像修復網絡和/或所述第二圖像修復網絡中使用ELU函數或PRLEU函數作為激活函數。