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

在審

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

本公開涉及一種圖像處理方法及裝置、電子設備和存儲介質，所述方法包括：對待處理圖像進行特征提取處理，獲得第一特征圖；根據第一特征圖，對待處理圖像中的目標對象進行邊緣預測處理，獲得目標對象的邊緣預測結果；根據邊緣預測結果以及第一特征圖，對待處理圖像中的目標對象進行分割處理，獲得目標對象的分割結果。根據本公開的實施例的圖像處理方法，可對待處理圖像中的目標對象進行邊緣預測處理，以檢測到目標對象的邊緣，并利用目標對象的邊緣來增強目標對象的分割輪廓，可提升分割目標的準確率。

技術領域

本公開涉及計算機技術領域，尤其涉及一種圖像處理方法及裝置、電子設備和存儲介質。

背景技術

在圖像處理中，對圖像中的目標對象的識別和分割具有重要作用。然而，在目標對象是透明物體或半透明物體的情況下，目標對象的透光度較高，可透過目標對象觀察到背景，則目標對象的外觀紋理與背景相似，因此，難以觀察和識別目標對象。

發明內容

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

根據本公開的一方面，提供了一種圖像處理方法，包括：對待處理圖像進行特征提取處理，獲得第一特征圖；根據所述第一特征圖，對所述待處理圖像中的目標對象進行邊緣預測處理，獲得所述目標對象的邊緣預測結果，其中，所述邊緣預測結果用于表示所述目標對象的邊緣；根據所述邊緣預測結果以及所述第一特征圖，對所述待處理圖像中的目標對象進行分割處理，獲得所述目標對象的分割結果。

根據本公開的實施例的圖像處理方法，可對待處理圖像中的目標對象進行邊緣預測處理，以檢測到目標對象的邊緣，并利用目標對象的邊緣來增強目標對象的分割輪廓，可提升分割目標的準確率。

在一種可能的實現方式中，對待處理圖像進行特征提取處理，獲得第一特征圖，包括：通過N個特征提取層級對待處理圖像進行特征提取處理，獲得分別與所述N個特征提取層級對應的第一特征圖，其中，N為大于1的整數。

在一種可能的實現方式中，根據所述第一特征圖，對所述待處理圖像中的目標對象進行邊緣預測處理，獲得所述目標對象的邊緣預測結果，包括：對第N個層級的第一特征圖進行邊緣特征提取，獲得邊緣特征圖；對所述邊緣特征圖以及所述N個層級中至少一個層級的第一特征圖進行解碼處理，獲得邊緣預測結果。

通過這種方式，可通過對邊緣特征圖以及至少一個層級的第一特征圖進行解碼處理，以通過融合具有低維信息的第一特征圖來增強邊緣特征圖中的邊緣特征，提升邊緣預測的準確率。

在一種可能的實現方式中，對所述邊緣特征圖以及所述N個層級中至少一個層級的第一特征圖進行解碼處理，獲得邊緣預測結果，包括：對所述邊緣特征圖進行升維處理，獲得第一升維特征圖；將所述第一升維特征圖與所述第一特征圖進行拼接處理，獲得第一拼接特征圖；對所述第一拼接特征圖進行卷積處理，獲得所述邊緣預測結果。

在一種可能的實現方式中，根據所述邊緣預測結果以及所述第一特征圖，對所述待處理圖像中的目標對象進行分割處理，獲得所述目標對象的分割結果，包括：對所述第一特征圖進行分割特征提取，獲得分割特征圖；根據所述邊緣預測結果和所述分割特征圖，獲得邊緣增強特征圖；對所述邊緣增強特征圖進行解碼處理，獲得所述分割結果。

通過這種方式，可通過邊緣預測結果對分割特征圖進行邊緣增強處理，以增強邊緣位置處的像素點，并將注意力機制集中于邊緣所在位置，提升分割結果的準確度。