[發(fā)明專利]訓(xùn)練數(shù)據(jù)增強方法、網(wǎng)絡(luò)模型訓(xùn)練方法、裝置和設(shè)備

在審

• 申請?zhí)枺?/td>: 202011228589.2
• 申請日: 2020-11-06
• 公開（公告）號: CN112418007A
• 公開（公告）日: 2021-02-26
• 發(fā)明（設(shè)計）人: 戚風(fēng)亮
• 申請（專利權(quán)）人: 上海眼控科技股份有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N20/00
• 代理公司: 廣州華進聯(lián)合專利商標(biāo)代理有限公司 44224
• 代理人: 姜曉云

本申請涉及計算機技術(shù)領(lǐng)域，特別是涉及一種訓(xùn)練數(shù)據(jù)增強方法、網(wǎng)絡(luò)模型訓(xùn)練方法、裝置和設(shè)備。所述方法包括：獲取初始訓(xùn)練數(shù)據(jù)，并識別初始訓(xùn)練數(shù)據(jù)中的至少兩個對象；獲取每個對象對應(yīng)的對象序列集，并從各對象序列集中確定對應(yīng)對象的目標(biāo)對象序列；分別從每個對象對應(yīng)的目標(biāo)對象序列中獲取待交換圖像；將所獲取的不同對象的待交換圖像進行交換，得到數(shù)據(jù)增強后的目標(biāo)訓(xùn)練數(shù)據(jù)。采用本方法能夠提升訓(xùn)練得到的模型的泛化能力。

技術(shù)領(lǐng)域

本申請涉及計算機技術(shù)領(lǐng)域，特別是涉及一種訓(xùn)練數(shù)據(jù)增強方法、網(wǎng)絡(luò)模型訓(xùn)練方法、裝置和設(shè)備。

背景技術(shù)

行人重識別(Person Re-identification，ReID)任務(wù)作為計算機視覺中的一個重要分支，在智慧城市，智慧交通等領(lǐng)域存在廣泛應(yīng)用。在這類任務(wù)中，一個移動目標(biāo)可能先后出現(xiàn)在不同的攝像頭中，利用計算機視覺的相關(guān)方法來判斷不同攝像頭中的兩個目標(biāo)是否屬于同一個目標(biāo)。

在傳統(tǒng)方式中，可以通過深度學(xué)習(xí)的方式進行ReID任務(wù)分析。但是由于對象移動的不穩(wěn)定性，很難有充足的訓(xùn)練數(shù)據(jù)進行模型訓(xùn)練，存在部分訓(xùn)練集中未涵蓋真實數(shù)據(jù)的缺陷，從而使得模型的泛化能力不足。

發(fā)明內(nèi)容

基于此，有必要針對上述技術(shù)問題，提供一種能夠提升模型泛化能力的訓(xùn)練數(shù)據(jù)增強方法、網(wǎng)絡(luò)模型訓(xùn)練方法、裝置和設(shè)備。

一種訓(xùn)練數(shù)據(jù)增強方法，訓(xùn)練數(shù)據(jù)用于機器學(xué)習(xí)，所述方法包括：

獲取初始訓(xùn)練數(shù)據(jù)，并識別初始訓(xùn)練數(shù)據(jù)中的至少兩個對象；

獲取每個對象對應(yīng)的對象序列集，并從各對象序列集中確定對應(yīng)對象的目標(biāo)對象序列；

分別從每個對象對應(yīng)的目標(biāo)對象序列中獲取待交換圖像；

將所獲取的不同對象的待交換圖像進行交換，得到數(shù)據(jù)增強后的目標(biāo)訓(xùn)練數(shù)據(jù)。

在其中一個實施例中，獲取初始訓(xùn)練數(shù)據(jù)，并識別初始訓(xùn)練數(shù)據(jù)中的至少兩個對象，包括：

對獲取的每個對象執(zhí)行身份檢測，直到獲取到至少兩個不同身份的對象為止。

在其中一個實施例中，目標(biāo)對象序列包括其對應(yīng)的對象的多幀圖像；

獲取每個對象對應(yīng)的對象序列集，并從各對象序列集中確定對應(yīng)對象的目標(biāo)對象序列，包括：

從初始訓(xùn)練數(shù)據(jù)中獲取第一對象對應(yīng)的第一對象序列集，并從第一對象序列集中獲取第一目標(biāo)對象序列；

從初始訓(xùn)練數(shù)據(jù)中獲取第二對象對應(yīng)的第二對象序列集，并從第二對象序列集中獲取第二目標(biāo)對象序列；

其中，第一對象序列集中包括對應(yīng)第一對象的k個第一對象序列，各第一對象序列中包括其對應(yīng)的第一對象的m幀圖像，第二對象序列集中包括對應(yīng)第二對象的f個第二對象序列，各第二對象序列中包括其對應(yīng)的第二對象的n幀圖像。

在其中一個實施例中，分別從每個對象對應(yīng)的目標(biāo)對象序列中獲取待交換圖像，包括：

從第一目標(biāo)對象序列中獲取第一待交換圖像；

從第二目標(biāo)對象序列中獲取第二待交換圖像；

將所獲取的不同對象的待交換圖像進行交換，得到數(shù)據(jù)增強后的目標(biāo)訓(xùn)練數(shù)據(jù)，包括：

將所獲取的第一待交換圖像和第二待交換圖像進行交換，得到數(shù)據(jù)增強后的目標(biāo)訓(xùn)練數(shù)據(jù)。

在其中一個實施例中，將所獲取的不同對象的待交換圖像進行交換，得到數(shù)據(jù)增強后的目標(biāo)訓(xùn)練數(shù)據(jù)，包括：

將所獲取的不同對象的待交換圖像進行交換，得到交換后的初始訓(xùn)練數(shù)據(jù)；

對交換后的初始訓(xùn)練數(shù)據(jù)進行通道轉(zhuǎn)換，生成預(yù)設(shè)通道數(shù)量的初始訓(xùn)練數(shù)據(jù)；