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

在審

• 申請號: 202210047520.2
• 申請日: 2022-01-17
• 公開（公告）號: CN114511445A
• 公開（公告）日: 2022-05-17
• 發明（設計）人: 陳博昱;高原;白錦峰
• 申請（專利權）人: 北京世紀好未來教育科技有限公司
• 主分類號: G06T3/40
• 分類號: G06T3/40;G06T9/00;G06N3/04;G06K9/62;G06V10/774;G06V10/82
• 代理公司: 北京開陽星知識產權代理有限公司 11710
• 代理人: 武玉維

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

獲取包含預設目標的待處理圖像以及與所述待處理圖像對應的參考背景圖像；

將所述參考背景圖像輸入至已訓練的空間變換網絡STN，獲得所述參考背景圖像的映射圖像；

將所述待處理圖像與所述映射圖像進行拼接，獲得輸入圖像；

將所述輸入圖像輸入至已訓練的編碼器-解碼器單元，獲得所述預設目標的透明圖像。

2.根據權利要求1所述的方法，其特征在于，所述已訓練的空間變換網絡STN包括級聯的卷積單元、平均池化單元和全連接單元；

所述卷積單元包括兩個級聯的卷積子單元，每個卷積子單元包括級聯的卷積層、最大池化層和激活函數層。

3.根據權利要求2所述的方法，其特征在于，所述將所述參考背景圖像輸入至已訓練的空間變換網絡STN，獲得所述參考背景圖像的映射圖像，包括：

將所述參考背景圖像輸入至所述級聯的卷積單元、平均池化單元和全連接單元，獲得仿射變換矩陣；

基于所述仿射變換矩陣對所述參考背景圖像進行映射，獲得所述映射圖像。

4.根據權利要求1所述的方法，其特征在于，所述已訓練的編碼器-解碼器單元包括級聯的預設編碼器、通道注意力子單元和預設解碼器。

5.根據權利要求4所述的方法，其特征在于，所述將所述輸入圖像輸入至已訓練的編碼器-解碼器單元，獲得所述預設目標的透明圖像，包括：

將所述輸入圖像輸入至所述預設編碼器，獲得編碼信息；

將所述編碼信息輸入至所述通道注意力子單元，獲得中間信息；

將所述中間信息輸入至所述預設解碼器，獲得所述透明圖像。

6.根據權利要求5所述的方法，其特征在于，所述通道注意力子單元包括：級聯的平均化層、第一卷積層、激活函數層、第二卷積層、歸一化層和疊加層。

7.根據權利要求6所述的方法，其特征在于，所述將所述編碼信息輸入至所述通道注意力子單元，獲得中間信息，包括：

將所述編碼信息輸入至所述級聯的平均化層、第一卷積層、激活函數層、第二卷積層和歸一化層，通過所述歸一化層輸出預設維度的信息；

所述疊加層用于對所述預設維度的信息與所述編碼信息做乘法運算，獲得所述中間信息。

8.一種圖像處理裝置，其特征在于，所述裝置包括：

獲取模塊，用于獲取包含預設目標的待處理圖像以及與所述待處理圖像對應的參考背景圖像；

第一輸入模塊，用于將所述參考背景圖像輸入至已訓練的空間變換網絡STN，獲得所述參考背景圖像的映射圖像；

拼接模塊，用于將所述待處理圖像與所述映射圖像進行拼接，獲得輸入圖像；

第二輸入模塊，用于將所述輸入圖像輸入至已訓練的編碼器-解碼器單元，獲得所述預設目標的透明圖像。

9.一種電子設備，其特征在于，所述電子設備包括：

處理器；以及

存儲程序的存儲器，

其中，所述程序包括指令，所述指令在由所述處理器執行時使所述處理器執行根據權利要求1-7中任一項所述的方法。

10.一種存儲有計算機指令的非瞬時計算機可讀存儲介質，其特征在于，所述計算機指令用于使所述計算機執行根據權利要求1-7中任一項所述的方法。