[發明專利]對象分割方法、裝置、設備及存儲介質

在審

• 申請號: 202210107771.5
• 申請日: 2022-01-28
• 公開（公告）號: CN114494298A
• 公開（公告）日: 2022-05-13
• 發明（設計）人: 朱淵略
• 申請（專利權）人: 北京字跳網絡技術有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06T7/90;G06K9/62;G06V10/26;G06V10/762;G06V10/764
• 代理公司: 北京遠智匯知識產權代理有限公司 11659
• 代理人: 穆雪

本公開實施例公開了一種對象分割方法、裝置、設備及存儲介質。包括：對待分割圖像中的目標對象進行語義識別，獲得初始掩膜圖；基于所述初始掩膜圖確定所述待分割圖像中的初始目標對象區域；對所述初始目標對象區域中的像素點按照顏色值進行聚類處理，獲取所述目標對象的N個顏色分類；根據所述N個顏色分類和所述待分割圖像獲得N個差值圖；根據所述N個差值圖與所述初始掩膜圖確定目標掩膜圖；基于所述目標掩膜圖對所述待分割圖像中的所述目標對象進行分割。本公開實施例提供的對象分割方法，根據差值圖和初始掩膜圖確定目標掩膜圖，從而基于目標掩膜圖分割目標對象，可以實現圖像中對象的分割，防止對象的漏分割，并提高對象分割的準確性。

技術領域

本公開實施例涉及圖像處理技術領域，尤其涉及一種對象分割方法、裝置、設備及存儲介質。

背景技術

目前，對于天空分割有如下兩種實現方法：一種是采用卷積神經網絡的深度學習算法分割天空，該種方式對于分割的掩膜圖存在中間局部漏分割的情況；另一種基于顏色信息的傳統算法，該方法依賴天空的顏色進行分割，可能出現誤分割的現象。對于色差小的圖片，也可能失效。

發明內容

本公開實施例提供一種對象分割方法、裝置、設備及存儲介質，以實現圖像中對象的分割，可以防止對象的漏分割，并提高對象分割的準確性。

第一方面，本公開實施例提供了一種對象分割方法，包括：

對待分割圖像中的目標對象進行語義識別，獲得初始掩膜圖；

基于所述初始掩膜圖確定所述待分割圖像中的初始目標對象區域；

對所述初始目標對象區域中的像素點按照顏色值進行聚類處理，獲取所述目標對象的N個顏色分類；其中，N為大于或者等于1的正整數；

根據所述N個顏色分類和所述待分割圖像獲得N個差值圖；

根據所述N個差值圖與所述初始掩膜圖確定目標掩膜圖；

基于所述目標掩膜圖對所述待分割圖像中的所述目標對象進行分割。

第二方面，本公開實施例還提供了一種對象分割裝置，包括：

初始掩膜圖獲取模塊，用于對待分割圖像中的目標對象進行語義識別，獲得初始掩膜圖；

初始目標對象區域確定模塊，用于基于所述初始掩膜圖確定所述待分割圖像中的初始目標對象區域；

聚類模塊，用于對所述初始目標對象區域中的像素點按照顏色值進行聚類處理，獲取所述目標對象的N個顏色分類；其中，N為大于或者等于1的正整數；

差值圖獲取模塊，用于根據所述N個顏色分類和所述待分割圖像獲得N個差值圖；

目標掩膜圖獲取模塊，用于根據所述N個差值圖與所述初始掩膜圖確定目標掩膜圖；

圖像分割模塊，用于基于所述目標掩膜圖對所述待分割圖像中的所述目標對象進行分割。

第三方面，本公開實施例還提供了一種電子設備，所述電子設備包括：

一個或多個處理裝置；

存儲裝置，用于存儲一個或多個程序；

當所述一個或多個程序被所述一個或多個處理裝置執行，使得所述一個或多個處理裝置實現如本公開實施例所述的對象分割方法。

第四方面，本公開實施例還提供了一種計算機可讀介質，其上存儲有計算機程序，該程序被處理裝置執行時實現如本公開實施例所述的對象分割方法。