[發(fā)明專利]圖像分割模型訓(xùn)練方法、圖像分割方法、裝置、設(shè)備及介質(zhì)

在審

• 申請?zhí)枺?/td>: 202010843033.8
• 申請日: 2020-08-20
• 公開（公告）號: CN111932546A
• 公開（公告）日: 2020-11-13
• 發(fā)明（設(shè)計）人: 宋苗;張海濤
• 申請（專利權(quán)）人: 展訊通信（上海）有限公司
• 主分類號: G06T7/10
• 分類號: G06T7/10
• 代理公司: 上海弼興律師事務(wù)所 31283
• 代理人: 楊東明;余中燕

1.一種圖像分割模型訓(xùn)練方法，其特征在于，包括：

獲取樣本數(shù)據(jù)集，所述樣本數(shù)據(jù)集包含若干訓(xùn)練圖像，各所述訓(xùn)練圖像分別標注有對應(yīng)的分割標簽；

根據(jù)所述樣本數(shù)據(jù)集對預(yù)先建立的圖像分割模型進行訓(xùn)練，所述圖像分割模型包含編碼器和解碼器，其中，所述解碼器采用子像素卷積網(wǎng)絡(luò)進行上采樣。

2.根據(jù)權(quán)利要求1所述的的圖像分割模型訓(xùn)練方法，其特征在于，在獲取樣本數(shù)據(jù)集后，所述方法還包括：

對所述樣本數(shù)據(jù)集進行數(shù)據(jù)增強處理。

3.根據(jù)權(quán)利要求1所述的的圖像分割模型訓(xùn)練方法，其特征在于，所述編碼器包含多個依次級聯(lián)的編碼模塊，其中，不同所述編碼模塊用于提取不同尺寸的編碼特征圖像。

4.根據(jù)權(quán)利要求3所述的的圖像分割模型訓(xùn)練方法，其特征在于，所述解碼器包含多個依次級聯(lián)的解碼模塊，其中，不同所述解碼模塊以跳躍連接方式與對應(yīng)所述編碼模塊相連，并且各所述解碼模塊分別包含所述子像素卷積模塊以及與所述子像素卷積模塊連接的殘差塊。

5.根據(jù)權(quán)利要求1所述的的圖像分割模型訓(xùn)練方法，其特征在于，所述編碼器采用輕量級網(wǎng)絡(luò)。

6.根據(jù)權(quán)利要求1所述的的圖像分割模型訓(xùn)練方法，其特征在于，所述子像素卷積模塊包含卷積層、重塑層和轉(zhuǎn)置層。

7.根據(jù)權(quán)利要求1所述的的圖像分割模型訓(xùn)練方法，其特征在于，所述根據(jù)所述樣本數(shù)據(jù)集對預(yù)先建立的圖像分割模型進行訓(xùn)練，包括：

從所述樣本數(shù)據(jù)集中選取目標訓(xùn)練圖像輸入至所述圖像分割模型中進行處理，得到所述目標訓(xùn)練圖像的分割結(jié)果；

根據(jù)所述目標訓(xùn)練圖像對應(yīng)的分割結(jié)果以及標注的分割標簽，確定損失函數(shù)；

根據(jù)所述損失函數(shù)對所述圖像分割模型進行迭代訓(xùn)練，直至滿足預(yù)定訓(xùn)練終止條件。

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

獲取待分割圖像；

基于權(quán)利要求1-7中任一項所述的方法訓(xùn)練得到的所述圖像分割模型對所述待分割圖像進行處理，得到所述待分割圖像的目標分割結(jié)果。

9.一種圖像分割模型訓(xùn)練裝置，其特征在于，包括：

樣本獲取模塊，用于獲取樣本數(shù)據(jù)集，所述樣本數(shù)據(jù)集包含若干訓(xùn)練圖像，各所述訓(xùn)練圖像分別標注有對應(yīng)的分割標簽；

模型訓(xùn)練模塊，用于根據(jù)所述樣本數(shù)據(jù)集對預(yù)先建立的圖像分割模型進行訓(xùn)練，所述圖像分割模型包含編碼器和解碼器，其中，所述解碼器采用子像素卷積網(wǎng)絡(luò)進行上采樣。

10.根據(jù)權(quán)利要求9所述的的圖像分割模型訓(xùn)練裝置，其特征在于，所述裝置還包括：數(shù)據(jù)增強模塊，用于在獲取樣本數(shù)據(jù)集后，對所述樣本數(shù)據(jù)集進行數(shù)據(jù)增強處理。

11.根據(jù)權(quán)利要求9所述的的圖像分割模型訓(xùn)練裝置，其特征在于，所述編碼器包含多個依次級聯(lián)的編碼模塊，其中，不同所述編碼模塊用于提取不同尺寸的編碼特征圖像。

12.根據(jù)權(quán)利要求11所述的的圖像分割模型訓(xùn)練裝置，其特征在于，所述解碼器包含多個依次級聯(lián)的解碼模塊，其中，不同所述解碼模塊以跳躍連接方式與對應(yīng)所述編碼模塊相連，并且各所述解碼模塊分別包含所述子像素卷積模塊以及與所述子像素卷積模塊連接的殘差塊。

13.根據(jù)權(quán)利要求9所述的的圖像分割模型訓(xùn)練裝置，其特征在于，所述編碼器采用輕量級網(wǎng)絡(luò)。

14.根據(jù)權(quán)利要求9所述的的圖像分割模型訓(xùn)練裝置，其特征在于，所述子像素卷積模塊包含卷積層、重塑層和轉(zhuǎn)置層。