[發明專利]圖像分割方法、裝置、電子設備和計算機可讀介質

在審

• 申請號: 202011083931.4
• 申請日: 2020-10-12
• 公開（公告）號: CN112215853A
• 公開（公告）日: 2021-01-12
• 發明（設計）人: 李華夏
• 申請（專利權）人: 北京字節跳動網絡技術有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06K9/00;G06K9/34;G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京衛智暢科專利代理事務所(普通合伙) 11557
• 代理人: 陳佳

1.一種圖像分割方法，包括：

獲取待處理圖像；

對所述待處理圖像進行特征提取，得到圖像特征；

基于所述圖像特征確定所述待處理圖像中是否包含目標對象；

響應于確定所述待處理圖像中包含所述目標對象，對所述待處理圖像進行分割，得到分割結果。

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

將所述待處理圖像輸入預先訓練的圖像分割網絡中的特征提取子網絡，得到所述圖像特征。

3.根據權利要求2所述的方法，其中，所述圖像分割網絡還包括分類子網絡；以及

所述基于所述圖像特征確定所述待處理圖像中是否包含目標對象，包括：

將所述圖像特征輸入所述分類子網絡，得到分類結果，所述分類結果用于表征所述待處理圖像中是否包含所述目標對象。

4.根據權利要求3所述的方法，其中，所述分類子網絡包括池化層和全連接層；以及

所述將所述圖像特征輸入所述分類子網絡，得到分類結果，包括：

將所述圖像特征輸入所述池化層，得到池化圖像特征；

將所述池化圖像特征輸入所述全連接層；

將所述全連接的輸出結果輸入激活函數，得到所述分類結果。

5.根據權利要求3所述的方法，其中，所述圖像分割網絡的訓練樣本包括樣本圖像、樣本分類結果和樣本分割結果；以及

所述圖像分割網絡在訓練過程中的損失值包括分類損失值和分割損失值，其中，所述分類損失值用于表征將所述樣本圖像輸入所述圖像分割網絡所得到的分類結果與所述樣本分類結果之間的差異，所述分割損失值用于表征將所述樣本圖像輸入所述圖像分割網絡所得到的分割結果與所述樣本分割結果之間的差異。

6.根據權利要求1所述的方法，其中，所述響應于確定所述待處理圖像中包含所述目標對象，對所述待處理圖像進行分割，得到分割結果，包括：

基于所述圖像特征，對所述待處理圖像進行分割，得到分割結果。

7.根據權利要求1-6中任一所述的方法，其中，所述方法還包括：

響應于確定所述待處理圖像中不包含所述目標對象，輸出用于表征所述待處理圖像中不包含所述目標對象的提示信息。

8.一種圖像分割裝置，包括：

獲取單元，被配置成獲取待處理圖像；

特征提取單元，被配置成對所述待處理圖像進行特征提取，得到圖像特征；

確定單元，被配置成基于所述圖像特征確定所述待處理圖像中是否包含目標對象；

分割單元，被配置成響應于確定所述待處理圖像中包含所述目標對象，對待處理圖像進行分割，得到分割結果。

9.一種電子設備，包括：

一個或多個處理器；

存儲裝置，其上存儲有一個或多個程序，

當所述一個或多個程序被所述一個或多個處理器執行，使得所述一個或多個處理器實現如權利要求1-7中任一所述的方法。

10.一種計算機可讀介質，其上存儲有計算機程序，其中，所述程序被處理器執行時實現如權利要求1-7中任一所述的方法。