[發(fā)明專利]行人重識別網(wǎng)絡的訓練及基于其的行人重識別

有效

• 申請?zhí)枺?/td>: 201710906719.5
• 申請日: 2017-09-29
• 公開（公告）號: CN108875487B
• 公開（公告）日: 2021-06-15
• 發(fā)明（設計）人: 羅浩;張弛
• 申請（專利權）人: 北京曠視科技有限公司;北京邁格威科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62
• 代理公司: 北京市磐華律師事務所 11336
• 代理人: 高偉;卜璐璐

1.一種行人重識別網(wǎng)絡的訓練方法，其特征在于，所述訓練方法包括：

利用分類損失對基準網(wǎng)絡進行預訓練；以及

聯(lián)合分類損失和五元組損失對經(jīng)預訓練的基準網(wǎng)絡進行調(diào)優(yōu)以得到行人重識別網(wǎng)絡；

所述聯(lián)合分類損失和五元組損失對經(jīng)預訓練的基準網(wǎng)絡進行調(diào)優(yōu)包括：

按預定要求和順序輸入五元組的五張樣本圖片；

基于所述基準網(wǎng)絡針對每張所述樣本圖片輸出的預測向量計算分類損失；

基于所述基準網(wǎng)絡針對所述五張樣本圖片輸出的特征向量計算五元組損失；以及

基于所計算的分類損失和所計算的五元組損失計算最終的損失以作為所述行人重識別網(wǎng)絡的損失。

2.根據(jù)權利要求1所述的訓練方法，其特征在于，所述利用分類損失對基準網(wǎng)絡進行預訓練包括：

將樣本圖片輸入到所述基準網(wǎng)絡；

將所述基準網(wǎng)絡針對所述樣本圖片輸出的預測向量與所述樣本圖片的標簽向量進行比較以得到分類損失；

基于所述分類損失調(diào)整所述基準網(wǎng)絡的參數(shù)；以及

反復進行上述步驟，直到分類準確度和分類損失基本不再變化。

3.根據(jù)權利要求2所述的訓練方法，其特征在于，所述基準網(wǎng)絡為殘差網(wǎng)絡。

4.根據(jù)權利要求3所述的訓練方法，其特征在于，在將所述樣本圖片輸入到所述基準網(wǎng)絡之前，對所述樣本圖片實施預處理操作。

5.根據(jù)權利要求1所述的訓練方法，其特征在于，所述所計算的分類損失為所述五張樣本圖片的分類損失的平均值。

6.根據(jù)權利要求1所述的訓練方法，其特征在于，所述五元組損失定義為：

l_qt＝d(正樣本1，正樣本2)-d(負樣本1，負樣本21)+d(負樣本21，負樣本22)-d(負樣本1，正樣本2)+a

其中，l_qt為五元組損失；正樣本1、正樣本2、負樣本1、負樣本21以及負樣本22為所述五張樣本圖片，且正樣本1和正樣本2為第一行人的兩張不同圖片，負樣本1為第二行人的圖片，負樣本21和負樣本22為第三行人的兩張不同圖片；d為兩張圖片的特征向量之間的距離；a為根據(jù)需求設置的常數(shù)參數(shù)。

7.根據(jù)權利要求1-6中的任一項所述的訓練方法，其特征在于，所述最終的損失為所述所計算的分類損失和所述所計算的五元組損失的加權和。

8.一種行人重識別網(wǎng)絡的訓練裝置，其特征在于，所述訓練裝置包括：

預訓練模塊，用于利用分類損失對基準網(wǎng)絡進行預訓練；以及

調(diào)優(yōu)模塊，用于聯(lián)合分類損失和五元組損失對經(jīng)預訓練的基準網(wǎng)絡進行調(diào)優(yōu)以得到行人重識別網(wǎng)絡；

所述調(diào)優(yōu)模塊對經(jīng)預訓練的基準網(wǎng)絡的調(diào)優(yōu)包括：

按預定要求和順序輸入五元組的五張樣本圖片；

基于所述基準網(wǎng)絡針對每張所述樣本圖片輸出的預測向量計算分類損失；

基于所述基準網(wǎng)絡針對所述五張樣本圖片輸出的特征向量計算五元組損失；以及

基于所計算的分類損失和所計算的五元組損失計算最終的損失以作為所述行人重識別網(wǎng)絡的損失。

9.根據(jù)權利要求8所述的訓練裝置，其特征在于，所述預訓練模塊對所述基準網(wǎng)絡的預訓練進一步包括：

將樣本圖片輸入到所述基準網(wǎng)絡；

將所述基準網(wǎng)絡針對所述樣本圖片輸出的預測向量與所述樣本圖片的標簽向量進行比較以得到分類損失；

基于所述分類損失調(diào)整所述基準網(wǎng)絡的參數(shù)；以及

反復進行上述操作，直到分類準確度和分類損失基本不再變化。

10.根據(jù)權利要求9所述的訓練裝置，其特征在于，所述基準網(wǎng)絡為殘差網(wǎng)絡。

11.根據(jù)權利要求10所述的訓練裝置，其特征在于，所述預訓練模塊還用于：在將所述樣本圖片輸入到所述基準網(wǎng)絡之前，對所述樣本圖片實施預處理操作。