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

在審

• 申請號: 202010231750.5
• 申請日: 2020-03-27
• 公開（公告）號: CN111445493A
• 公開（公告）日: 2020-07-24
• 發明（設計）人: 謝恩澤;羅平;王文佳
• 申請（專利權）人: 北京市商湯科技開發有限公司
• 主分類號: G06T7/13
• 分類號: G06T7/13;G06T7/00;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

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

對待處理圖像進行特征提取處理，獲得第一特征圖；

根據所述第一特征圖，對所述待處理圖像中的目標對象進行邊緣預測處理，獲得所述目標對象的邊緣預測結果，其中，所述邊緣預測結果用于表示所述目標對象的邊緣；

根據所述邊緣預測結果以及所述第一特征圖，對所述待處理圖像中的目標對象進行分割處理，獲得所述目標對象的分割結果。

2.根據權利要求1所述的方法，其特征在于，對待處理圖像進行特征提取處理，獲得第一特征圖，包括：

通過N個特征提取層級對待處理圖像進行特征提取處理，獲得分別與所述N個特征提取層級對應的第一特征圖，其中，N為大于1的整數。

3.根據權利要求2所述的方法，其特征在于，根據所述第一特征圖，對所述待處理圖像中的目標對象進行邊緣預測處理，獲得所述目標對象的邊緣預測結果，包括：

對第N個層級的第一特征圖進行邊緣特征提取，獲得邊緣特征圖；

對所述邊緣特征圖以及所述N個層級中至少一個層級的第一特征圖進行解碼處理，獲得邊緣預測結果。

4.根據權利要求3所述的方法，其特征在于，對所述邊緣特征圖以及所述N個層級中至少一個層級的第一特征圖進行解碼處理，獲得邊緣預測結果，包括：

對所述邊緣特征圖進行升維處理，獲得第一升維特征圖；

將所述第一升維特征圖與所述第一特征圖進行拼接處理，獲得第一拼接特征圖；

對所述第一拼接特征圖進行卷積處理，獲得所述邊緣預測結果。

5.根據權利要求1-4任一項所述的方法，其特征在于，根據所述邊緣預測結果以及所述第一特征圖，對所述待處理圖像中的目標對象進行分割處理，獲得所述目標對象的分割結果，包括：

對所述第一特征圖進行分割特征提取，獲得分割特征圖；

根據所述邊緣預測結果和所述分割特征圖，獲得邊緣增強特征圖；

對所述邊緣增強特征圖進行解碼處理，獲得所述分割結果。

6.根據權利要求5所述的方法，其特征在于，所述第一特征圖包括通過N個特征提取層級進行特征提取處理得到的N個第一特征圖；所述邊緣增強特征圖包括與分割特征圖對應的第一邊緣增強特征圖，以及與N個層級中的至少一個特征提取層級的第一特征圖對應的第二邊緣增強特征圖，

其中，根據所述邊緣預測結果和所述分割特征圖，獲得邊緣增強特征圖，包括：

通過所述邊緣預測結果對所述分割特征圖進行第一邊緣增強處理，獲得第一邊緣增強特征圖；

通過所述邊緣預測結果對N個層級中的至少一個特征提取層級的第一特征圖分別進行第二邊緣增強處理，獲得與所述至少一個特征提取層級的第一特征圖分別對應的第二邊緣增強特征圖。

7.根據權利要求6所述的方法，其特征在于，所述邊緣預測結果包括邊緣預測圖，

其中，通過所述邊緣預測結果對所述分割特征圖進行第一邊緣增強處理，獲得第一邊緣增強特征圖，包括：

對所述邊緣預測圖和所述分割特征圖中的對應像素點的像素值相乘，獲得第二特征圖；

對所述第二特征圖和所述邊緣預測圖進行拼接處理，獲得第三特征圖；

對所述第三特征圖進行邊緣增強處理，獲得所述第一邊緣增強特征圖。

8.根據權利要求7所述的方法，其特征在于，對所述第三特征圖進行邊緣增強處理，獲得所述第一邊緣增強特征圖，包括：

將所述第三特征圖進行深度卷積處理，獲得深度卷積結果；

將所述深度卷積結果進行特征提取處理，獲得特征提取結果；

將所述特征提取結果與所述深度卷積結果的對應像素點進行相乘處理，獲得所述第一邊緣增強特征圖。