[發明專利]一種圖像背景替換方法、系統、電子設備及存儲介質

在審

• 申請號: 202110740641.0
• 申請日: 2021-06-30
• 公開（公告）號: CN113409188A
• 公開（公告）日: 2021-09-17
• 發明（設計）人: 李艷宇;羅濤;施佳子;于海燕
• 申請（專利權）人: 中國工商銀行股份有限公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06T7/12;G06T7/194;G06N3/04
• 代理公司: 中科專利商標代理有限責任公司 11021
• 代理人: 周天宇

1.一種圖像背景替換方法，其特征在于，包括：

獲取第一圖像中每個像素點的前景預測概率；

選取所述前景預測概率接近特征值的點作為不確定點；

將所述不確定點的不同尺度的特征圖進行融合，得到邊界清晰的前景圖和α遮罩；

使用待替換的背景圖像與所述前景圖和α遮罩進行合成，得到新背景下的第二圖像。

2.根據權利要求1所述的圖像背景替換方法，其特征在于，所述獲取第一圖像中每個像素點的前景預測概率包括：

獲取第一圖像；

將所述第一圖像輸入卷積神經網絡得到該圖像的特征圖；

將所述特征圖輸入金字塔池化模塊，以獲得每個像素點的前景預測概率。

3.根據權利要求2所述的圖像背景替換方法，其特征在于，所述將所述特征圖輸入金字塔池化模塊包括：

構建多層的特征金字塔，

將所述特征圖輸入各層金字塔進行不同尺度的池化；

進行上采樣、全連接層、激活函數處理后，得到每個像素點的前景預測概率。

4.根據權利要求1所述的圖像背景替換方法，其特征在于，所述選取所述前景預測概率接近特征值的點作為不確定點還包括：

反卷積處理，得到粗糙分割的特征圖。

5.根據權利要求4所述的圖像背景替換方法，其特征在于，所述選取所述前景預測概率接近特征值的點作為不確定點包括：

采用雙線性插值對所述粗糙分割的特征圖的邊界進行上采樣；

獲取所述邊界的像素點的前景預測概率；

選擇概率接近特征值的點作為不確定點。

6.根據權利要求5所述的圖像背景替換方法，其特征在于，所述前景預測概率來于摳圖算法的公式：

I＝αF+(1-α)B

其中I為圖像，F為I的前景，α為像素的前景預測概率，B為I的背景。

7.根據權利要求6所述的圖像背景替換方法，其特征在于，所述選取所述前景預測概率接近特征值的點作為不確定點中所述特征值為0.5。

8.根據權利要求3所述的圖像背景替換方法，其特征在于，所述將所述不確定點的不同尺度的特征圖進行融合，得到邊界清晰的前景圖和α遮罩包括：

將所述特征圖輸入各層金字塔進行不同尺度的池化，得到不同尺度的特征圖；

將所述不確定點的不同尺度的特征圖進行融合，得到邊界清晰的特征圖。

9.根據權利要求8所述的圖像背景替換方法，其特征在于，所述將所述不確定點的不同尺度的特征圖進行融合還包括：

解碼，分別得到邊界清晰的前景圖和α遮罩。

10.根據權利要求9所述的圖像背景替換方法，其特征在于，所述解碼得到邊界清晰的前景圖包括：

雙線性差值上采樣、卷積、BN、ReLU處理，得到預測的特征圖；

對所述預測的特征圖和所述邊界清晰的特征圖進行雙線性差值上采樣、卷積、BN、ReLU、鏡面Padding、卷積，得到邊界清晰的前景圖。

11.根據權利要求9所述的圖像背景替換方法，其特征在于，所述解碼得到邊界清晰的α遮罩包括：

雙線性差值上采樣、卷積、BN、ReLU處理，進行解碼；

鏡面Padding、卷積、Tanh處理，得到邊界清晰的α遮罩。

12.根據權利要求1所述的圖像背景替換方法，其特征在于，所述使用待替換的背景圖像與所述前景圖和α遮罩進行合成包括：

將α遮罩與前景圖進行處理得到第二圖像的前景部分；

將α遮罩與待替換的背景圖像進行處理得到第二圖像的背景部分；

將所述前景部分與所述背景部分進行疊加得到第二圖像。