[發(fā)明專利]人體圖像分割方法及設備

有效

• 申請?zhí)枺?/td>: 201910601007.1
• 申請日: 2019-07-04
• 公開（公告）號: CN110310293B
• 公開（公告）日: 2021-08-10
• 發(fā)明（設計）人: 李華夏
• 申請（專利權）人: 北京字節(jié)跳動網(wǎng)絡技術有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06T7/194;G06K9/00;G06K9/34
• 代理公司: 上海光柵知識產(chǎn)權代理有限公司 31340
• 代理人: 馬雯雯

1.一種人體圖像分割方法，其特征在于，包括：

將原始圖像輸入至預設神經(jīng)網(wǎng)絡，獲取所述預設神經(jīng)網(wǎng)絡輸出的第一人體分割圖像，其中所述第一人體分割圖像中的人體分割區(qū)域像素值為1，背景區(qū)域像素值為0；

將所述第一人體分割圖像的像素值乘以255，得到第二人體分割圖像，其中所述第二人體分割圖像中的人體分割區(qū)域的像素值為255，背景區(qū)域像素值為0；

對所述第二人體分割圖像進行雙三次插值，得到第三人體分割圖像，其中所述第三人體分割圖像的像素值為0至255；

將所述第三人體分割圖像的像素值除以255，得到目標分割掩膜，其中目標分割掩膜的像素值為0至1；

根據(jù)所述目標分割掩膜和所述原始圖像，得到人體分割區(qū)域的圖像。

2.根據(jù)權利要求1所述的方法，其特征在于，所述原始圖像為第一尺寸，所述預設神經(jīng)網(wǎng)絡輸入輸出圖像均為第二尺寸，其中第一尺寸大于第二尺寸；

所述將原始圖像輸入至預設神經(jīng)網(wǎng)絡，獲取所述預設神經(jīng)網(wǎng)絡輸出的第一人體分割圖像，包括：

對第一尺寸的原始圖像進行下采樣處理，得到第二尺寸的原始圖像；

將所述第二尺寸的原始圖像輸入至所述預設神經(jīng)網(wǎng)絡，獲取所述預設神經(jīng)網(wǎng)絡輸出的第二尺寸的第一人體分割圖像；

相應地，對所述第二人體分割圖像進行雙三次插值，得到第三人體分割圖像，包括：

對第二尺寸的第二人體分割圖像雙三次插值上采樣，得到第一尺寸的第三人體分割圖像。

3.根據(jù)權利要求1所述的方法，其特征在于，所述根據(jù)所述目標分割掩膜和所述原始圖像，得到人體分割區(qū)域的圖像，包括：

將所述原始圖像的像素值乘以(1-目標分割掩膜像素值)，得到人體分割區(qū)域的圖像。

4.根據(jù)權利要求1所述的方法，其特征在于，所述方法還包括：

將所述原始圖像的像素值乘以(1-目標分割掩膜像素值)，并加上選取的目標背景圖像與目標分割掩膜像素值的乘積，得到人體分割區(qū)域與目標背景圖像融合后的圖像。

5.根據(jù)權利要求2所述的方法，其特征在于，所述第一尺寸為1920×1080，所述第二尺寸為448×256。

6.一種人體圖像分割設備，其特征在于，包括：

第一人體分割圖像獲取單元，用于將原始圖像輸入至預設神經(jīng)網(wǎng)絡，獲取所述預設神經(jīng)網(wǎng)絡輸出的第一人體分割圖像，其中所述第一人體分割圖像中的人體分割區(qū)域像素值為1，背景區(qū)域像素值為0；

第二人體分割圖像獲取單元，用于將所述第一人體分割圖像的像素值乘以255，得到第二人體分割圖像，其中所述第二人體分割圖像中的人體分割區(qū)域的像素值為255，背景區(qū)域像素值為0；

第三人體分割圖像獲取單元，用于對所述第二人體分割圖像進行雙三次插值，得到第三人體分割圖像，其中所述第三人體分割圖像的像素值為0至255；

目標分割掩膜獲取單元，用于將所述第三人體分割圖像的像素值除以255，得到目標分割掩膜，其中目標分割掩膜的像素值為0至1；

人體分割區(qū)域的圖像取單元，用于根據(jù)所述目標分割掩膜和所述原始圖像，得到人體分割區(qū)域的圖像。

7.根據(jù)權利要求6所述的設備，其特征在于，所述原始圖像為第一尺寸，所述預設神經(jīng)網(wǎng)絡輸入輸出圖像均為第二尺寸，其中第一尺寸大于第二尺寸；

所述第一人體分割圖像獲取單元401，還用于對第一尺寸的原始圖像進行下采樣處理，得到第二尺寸的原始圖像；將所述第二尺寸的原始圖像輸入至所述預設神經(jīng)網(wǎng)絡，獲取所述預設神經(jīng)網(wǎng)絡輸出的第二尺寸的第一人體分割圖像；

相應地，所述第三人體分割圖像獲取單元403，還用于對第二尺寸的第二人體分割圖像雙三次插值上采樣，得到第一尺寸的第三人體分割圖像。

8.根據(jù)權利要求6所述的設備，其特征在于，所述人體分割區(qū)域的圖像取單元，具體用于將所述原始圖像的像素值乘以(1-目標分割掩膜像素值)，得到人體分割區(qū)域的圖像。