[發(fā)明專利]目標(biāo)檢測模型的訓(xùn)練方法、裝置、目標(biāo)檢測方法、裝置、介質(zhì)及設(shè)備

在審

• 申請?zhí)枺?/td>: 202210515401.5
• 申請日: 2022-05-11
• 公開（公告）號: CN114943872A
• 公開（公告）日: 2022-08-26
• 發(fā)明（設(shè)計）人: 易長渝
• 申請（專利權(quán)）人: 上海云從企業(yè)發(fā)展有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/82;G06V10/75;G06V40/10;G06N3/08
• 代理公司: 上海光華專利事務(wù)所(普通合伙) 31219
• 代理人: 張雙鳳

1.一種目標(biāo)檢測模型的訓(xùn)練方法，其特征在于，包括：

獲取包括多個對象的圖片；

對所述圖片進(jìn)行預(yù)處理；所述預(yù)處理包括：

對多個對象進(jìn)行標(biāo)注，得到多個對象的人頭真值框與人體真值框；對所述人頭真值框和所述人體真值框進(jìn)行匹配，確定屬于同一個對象的人頭真值框和人體真值框；基于所述人體真值框，得到與所述人體真值框匹配的軀干真值框，以完成對圖片的標(biāo)注；

將完成標(biāo)注的圖片作為樣本圖片對初始神經(jīng)網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練，得到目標(biāo)檢測模型。

2.根據(jù)權(quán)利要求1所述的目標(biāo)檢測模型的訓(xùn)練方法，其特征在于，所述對所述人頭真值框和所述人體真值框進(jìn)行匹配，包括：

通過所述人頭真值框與所述人體真值框的IOU以及位置關(guān)系對所述人頭真值框和所述人體真值框進(jìn)行匹配。

3.根據(jù)權(quán)利要求1所述的目標(biāo)檢測模型的訓(xùn)練方法，其特征在于，所述目標(biāo)檢測模型包括人頭檢測分支、人體檢測分支以及軀干檢測分支；所述軀干檢測分支的通道數(shù)為12*A，人頭檢測分支以及人體檢測分支的通道數(shù)為4*A，其中，A表示錨框個數(shù)。

4.根據(jù)權(quán)利要求3所述的目標(biāo)檢測模型的訓(xùn)練方法，其特征在于，所述軀干檢測分支以軀干真值框的區(qū)域、人頭真值框的區(qū)域、人體真值框的區(qū)域作為輸入，輸出人體檢測框、軀干檢測框和人頭檢測框。

5.根據(jù)權(quán)利要求1所述的目標(biāo)檢測模型的訓(xùn)練方法，其特征在于，采用NMS方法對所述軀干檢測框進(jìn)行去重。

6.一種目標(biāo)檢測模型的訓(xùn)練裝置，其特征在于，包括：

圖片獲取模塊，用于獲取包括多個對象的圖片；

圖片預(yù)處理模塊，用于對所述圖片進(jìn)行預(yù)處理；所述預(yù)處理模塊包括：

第一標(biāo)注子模塊，用于對多個對象進(jìn)行標(biāo)注，得到多個對象的人頭真值框與人體真值框；

匹配子模塊，用于對所述人頭真值框和所述人體真值框進(jìn)行匹配，確定屬于同一個對象的人頭真值框和人體真值框；

第二標(biāo)注子模塊，用于基于所述人體真值框，得到與所述人體真值框匹配的軀干真值框，以完成對圖片的標(biāo)注；

訓(xùn)練模塊，用于將完成標(biāo)注的圖片作為樣本圖片對初始神經(jīng)網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練，得到目標(biāo)檢測模型。

7.根據(jù)權(quán)利要求6所述的目標(biāo)檢測模型的訓(xùn)練裝置，其特征在于，所述匹配子模塊通過所述人頭真值框與所述人體真值框的IOU以及位置關(guān)系對所述人頭真值框和所述人體真值框進(jìn)行匹配。

8.根據(jù)權(quán)利要求6所述的目標(biāo)檢測模型的訓(xùn)練裝置，其特征在于，所述目標(biāo)檢測模型包括人頭檢測分支、人體檢測分支以及軀干檢測分支；所述軀干檢測分支的通道數(shù)為12*A，人頭檢測分支以及人體檢測分支的通道數(shù)為4*A，其中，A表示錨框個數(shù)；所述軀干檢測分支以軀干真值框的區(qū)域、人頭真值框的區(qū)域、人體真值框的區(qū)域作為輸入，輸出人體檢測框、軀干檢測框和人頭檢測框。

9.根據(jù)權(quán)利要求6所述的目標(biāo)檢測模型的訓(xùn)練裝置，其特征在于，還包括：去重模塊，用于采用NMS方法對所述軀干檢測框進(jìn)行去重。

10.一種目標(biāo)檢測方法，其特征在于，包括：

獲取待檢測圖片；

對所述待檢測圖片進(jìn)行特征提取，得到人體生物特征，所述人體生物特征包括：人頭特征、人體特征以及軀干特征；

基于所述生物特征，利用如權(quán)利要求1-5任意一項(xiàng)所述的目標(biāo)檢測模型對所述檢測圖片進(jìn)行檢測，得到目標(biāo)對象。

11.一種目標(biāo)檢測裝置，其特征在于，包括：

圖片獲取模塊，用于獲取待檢測圖片；

特征提取模塊，用于對所述待檢測圖片進(jìn)行特征提取，得到人體生物特征，所述人體生物特征包括：人頭特征、人體特征以及軀干特征；

目標(biāo)檢測模塊，用于基于所述生物特征，利用如權(quán)利要求1-5任意一項(xiàng)所述的目標(biāo)檢測模型對所述檢測圖片進(jìn)行檢測，得到目標(biāo)對象。