[發(fā)明專利]一種人像分割方法及系統(tǒng)

在審

• 申請?zhí)枺?/td>: 201811204098.7
• 申請日: 2018-10-16
• 公開（公告）號(hào): CN109523558A
• 公開（公告）日: 2019-03-26
• 發(fā)明（設(shè)計(jì)）人: 胡事民;張松海;董欣;李瑞龍
• 申請（專利權(quán)）人: 清華大學(xué)
• 主分類號(hào): G06T7/11
• 分類號(hào): G06T7/11
• 代理公司: 北京路浩知識(shí)產(chǎn)權(quán)代理有限公司 11002
• 代理人: 王瑩;吳歡燕

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

基于訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)，對人像圖像進(jìn)行分割，所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)是在卷積神經(jīng)網(wǎng)絡(luò)中添加depthwise卷積層構(gòu)造的。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述基于訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)，對人像圖像進(jìn)行分割，包括：

將所述訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)轉(zhuǎn)換為Caffe模型；

將所述Caffe模型轉(zhuǎn)換為Coreml模型；

將所述人像圖像輸入至所述Coreml模型中，以對所述人像圖像進(jìn)行分割。

3.根據(jù)權(quán)利要求1所述的方法，其特征在于，在所述基于訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)，對人臉圖像進(jìn)行分割之前，還包括：

對訓(xùn)練樣本集進(jìn)行預(yù)處理；

基于預(yù)處理后的訓(xùn)練樣本集，對所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述對訓(xùn)練樣本集進(jìn)行預(yù)處理，包括：

將所述訓(xùn)練樣本集中不含人像的圖像樣本標(biāo)記為負(fù)樣本集，并將所述訓(xùn)練樣本集中含有人像的圖像樣本標(biāo)記為正樣本集。

5.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述基于預(yù)處理后的訓(xùn)練樣本集，對所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練，包括：

基于Pytorch框架以及預(yù)處理后的訓(xùn)練樣本集，對所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練，得到Pytorch模型；

將所述Pytorch模型作為所述訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)。

6.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述基于預(yù)處理后的訓(xùn)練樣本集，對所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練，還包括：

將所述預(yù)處理后的訓(xùn)練樣本集輸入到所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)中，并在所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)中添加特征融合層。

7.根據(jù)權(quán)利要求1至6任一項(xiàng)所述的方法，其特征在于，在所述基于訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)，對人像圖像進(jìn)行分割之前，所述方法還包括：

在所述卷積神經(jīng)網(wǎng)絡(luò)中添加多個(gè)depthwise卷積層、多個(gè)上采樣層和多個(gè)下采樣層，以得到所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)。

8.一種人像分割系統(tǒng)，其特征在于，包括：

分割模塊，用于基于訓(xùn)練后的輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)，對人像圖像進(jìn)行分割，所述輕量級(jí)卷積神經(jīng)網(wǎng)絡(luò)是在卷積神經(jīng)網(wǎng)絡(luò)中添加depthwise卷積層構(gòu)造的。

9.一種人像分割設(shè)備，包括存儲(chǔ)器、處理器及存儲(chǔ)在存儲(chǔ)器上并可在處理器上運(yùn)行的計(jì)算機(jī)程序，其特征在于，所述處理器執(zhí)行所述程序時(shí)實(shí)現(xiàn)如權(quán)利要求1至7任一項(xiàng)所述人像分割方法。

10.一種非暫態(tài)計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其特征在于，所述非暫態(tài)計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)存儲(chǔ)計(jì)算機(jī)指令，所述計(jì)算機(jī)指令使所述計(jì)算機(jī)執(zhí)行如權(quán)利要求1至7任一項(xiàng)所述人像分割方法。