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

在審

• 申請號: 202011383926.5
• 申請日: 2020-11-30
• 公開（公告）號: CN112508964A
• 公開（公告）日: 2021-03-16
• 發明（設計）人: 彭冕;王健;孫昊;譚嘯;丁二銳
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11
• 代理公司: 北京銀龍知識產權代理有限公司 11243
• 代理人: 許靜;黃燦

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

獲取輸入人物圖像的第一分割概率圖；

檢測所述輸入人物圖像的目標部位所在區域，并獲取所述區域對應的包括所述目標部位的局部圖像；

獲取所述第一分割概率圖中所述區域的局部分割概率圖；

依據所述局部分割概率圖對所述局部圖像進行分割，得到第二分割概率圖；

將所述第一分割概率圖和所述第二分割概率圖進行結合，得到所述輸入人物圖像的分割結果。

2.如權利要求1所述的方法，其中，所述依據所述局部分割概率圖對所述局部圖像進行分割，得到第二分割概率圖，包括：

生成局部分割概率圖的三分圖；

將所述局部圖像和所述三分圖作為第一分割網絡的輸入進行分割，得到第二分割概率圖。

3.如權利要求1所述的方法，其中，所述將所述第一分割概率圖和所述第二分割概率圖進行結合，得到所述輸入人物圖像的分割結果，包括：

將所述第一分割概率圖中的所述區域的圖像內容替換為所述第二分割概率圖的圖像內容，得到所述輸入人物圖像的分割結果；或者

將所述第一分割概率圖中的所述區域的第一圖像內容替換為所述第二分割概率圖的第一圖像內容，將所述第一分割概率圖中的所述區域的第二圖像內容與所述第二分割概率圖的第二圖像內容進行融合，得到所述輸入人物圖像的分割結果，其中，所述第一分割概率圖的第一圖像內容在所述區域的位置與所述第二分割概率圖的第一圖像內容在所述第二分割概率圖中的位置相同，所述第二分割概率圖的第一圖像內容在所述區域的位置與所述第二分割概率圖的第二圖像內容在所述第二分割概率圖中的位置相同。

4.如權利要求1所述的方法，其中，所述第一分割概率圖為所述輸入人物圖像中各像素點為人體部位的概率圖，所述目標部位為頭部，所述第一分割概率圖為所述局部圖像中各像素點為頭部的概率圖。

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

第一獲取模塊，用于獲取輸入人物圖像的第一分割概率圖；

第二獲取模塊，用于檢測所述輸入人物圖像的目標部位所在區域，并獲取所述區域對應的包括所述目標部位的局部圖像；

第三獲取模塊，用于獲取所述第一分割概率圖中所述區域的局部分割概率圖；

分割模塊，用于依據所述局部分割概率圖對所述局部圖像進行分割，得到第二分割概率圖；

結合模塊，用于將所述第一分割概率圖和所述第二分割概率圖進行結合，得到所述輸入人物圖像的分割結果。

6.如權利要求5所述的裝置，其中，所述分割模塊，包括：

生成單元，用于生成局部分割概率圖的三分圖；

分割單元，用于將所述局部圖像和所述三分圖作為第一分割網絡的輸入進行分割，得到第二分割概率圖。

7.如權利要求5所述的裝置，其中，所述結合模塊用于將所述第一分割概率圖中的所述區域的圖像內容替換為所述第二分割概率圖的圖像內容，得到所述輸入人物圖像的分割結果；或者

所述結合模塊用于將所述第一分割概率圖中的所述區域的第一圖像內容替換為所述第二分割概率圖的第一圖像內容，將所述第一分割概率圖中的所述區域的第二圖像內容與所述第二分割概率圖的第二圖像內容進行融合，得到所述輸入人物圖像的分割結果，其中，所述第一分割概率圖的第一圖像內容在所述區域的位置與所述第二分割概率圖的第一圖像內容在所述第二分割概率圖中的位置相同，所述第二分割概率圖的第一圖像內容在所述區域的位置與所述第二分割概率圖的第二圖像內容在所述第二分割概率圖中的位置相同。

8.如權利要求5所述的裝置，其中，所述第一分割概率圖為所述輸入人物圖像中各像素點為人體部位的概率圖，所述目標部位為頭部，所述第一分割概率圖為所述局部圖像中各像素點為頭部的概率圖。