[發(fā)明專利]目標(biāo)重識(shí)別方法及其網(wǎng)絡(luò)訓(xùn)練方法、相關(guān)裝置

有效

• 申請(qǐng)?zhí)枺?/td>: 202010637098.7
• 申請(qǐng)日: 2020-07-03
• 公開(kāi)（公告）號(hào): CN111814655B
• 公開(kāi)（公告）日: 2023-09-01
• 發(fā)明（設(shè)計(jì)）人: 孫立波;潘華東;殷俊
• 申請(qǐng)（專利權(quán)）人: 浙江大華技術(shù)股份有限公司
• 主分類號(hào): G06V20/60
• 分類號(hào): G06V20/60;G06V40/10;G06V10/774
• 代理公司: 深圳市威世博知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44280
• 代理人: 黎堅(jiān)怡

本申請(qǐng)公開(kāi)了一種目標(biāo)重識(shí)別方法及其網(wǎng)絡(luò)訓(xùn)練方法、相關(guān)裝置。該方法包括：獲取訓(xùn)練圖像集，訓(xùn)練圖像集中包括多個(gè)目標(biāo)的圖像，且每個(gè)目標(biāo)至少對(duì)應(yīng)兩種風(fēng)格的圖像；利用目標(biāo)重識(shí)別網(wǎng)絡(luò)對(duì)訓(xùn)練圖像集中每張圖像進(jìn)行識(shí)別，得到每張圖像的識(shí)別結(jié)果，識(shí)別結(jié)果包括圖像的特征；基于識(shí)別結(jié)果獲取目標(biāo)重識(shí)別網(wǎng)絡(luò)的損失，損失包括第一距離損失，第一距離損失是基于目標(biāo)對(duì)應(yīng)的不同風(fēng)格的圖像的特征之間的距離得到的；基于損失調(diào)整所述目標(biāo)重識(shí)別網(wǎng)絡(luò)的參數(shù)；重復(fù)上述步驟直至滿足停止訓(xùn)練的條件。通過(guò)上述方式，本申請(qǐng)能夠提高目標(biāo)重識(shí)別網(wǎng)絡(luò)的目標(biāo)識(shí)別性能。

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

本申請(qǐng)涉及圖像識(shí)別技術(shù)領(lǐng)域，特別是涉及一種目標(biāo)重識(shí)別方法及其網(wǎng)絡(luò)訓(xùn)練方法、相關(guān)裝置。

背景技術(shù)

相對(duì)于視角較固定的人臉識(shí)別任務(wù)，行人重識(shí)別不受人臉視角、人體姿態(tài)和攝像機(jī)視角的限制，其識(shí)別效果相對(duì)更好、應(yīng)用場(chǎng)景更廣泛。因此，近年來(lái)目標(biāo)重識(shí)別技術(shù)取得了飛速的發(fā)展。

目標(biāo)重識(shí)別技術(shù)旨在從視頻或圖像中找到特定的目標(biāo)，其在安防、新零售和互聯(lián)網(wǎng)等領(lǐng)域具有重要的應(yīng)用價(jià)值。目標(biāo)重識(shí)別技術(shù)針對(duì)的目標(biāo)可以為行人、車輛等。一般采用目標(biāo)重識(shí)別網(wǎng)絡(luò)來(lái)進(jìn)行目標(biāo)識(shí)別，例如，使用行人重識(shí)別(Person?Re-identification,ReID)網(wǎng)絡(luò)來(lái)對(duì)行人圖像進(jìn)行識(shí)別。

現(xiàn)有技術(shù)中，在使用目標(biāo)重識(shí)別網(wǎng)絡(luò)之前，通常需要對(duì)其進(jìn)行訓(xùn)練，使其后續(xù)在使用過(guò)程中得到的識(shí)別結(jié)果更加準(zhǔn)確。然而，現(xiàn)有的訓(xùn)練方法得到的目標(biāo)重識(shí)別網(wǎng)絡(luò)對(duì)目標(biāo)的識(shí)別效果不夠好。

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

本申請(qǐng)?zhí)峁┮环N圖像增強(qiáng)方法、電子設(shè)備及存儲(chǔ)介質(zhì)，能夠解決現(xiàn)有的訓(xùn)練方法得到的目標(biāo)重識(shí)別網(wǎng)絡(luò)對(duì)目標(biāo)的識(shí)別效果不夠好的問(wèn)題。

為解決上述技術(shù)問(wèn)題，本申請(qǐng)采用的一個(gè)技術(shù)方案是：提供一種目標(biāo)重識(shí)別網(wǎng)絡(luò)的訓(xùn)練方法，該方法包括：獲取訓(xùn)練圖像集，訓(xùn)練圖像集中包括多個(gè)目標(biāo)的圖像，且每個(gè)目標(biāo)至少對(duì)應(yīng)兩種風(fēng)格的圖像；利用目標(biāo)重識(shí)別網(wǎng)絡(luò)對(duì)訓(xùn)練圖像集中每張圖像進(jìn)行識(shí)別，得到每張圖像的識(shí)別結(jié)果，識(shí)別結(jié)果包括圖像的特征；基于識(shí)別結(jié)果獲取目標(biāo)重識(shí)別網(wǎng)絡(luò)的損失，損失包括第一距離損失，第一距離損失是基于目標(biāo)對(duì)應(yīng)的不同風(fēng)格的圖像的特征之間的距離得到的；基于損失調(diào)整所述目標(biāo)重識(shí)別網(wǎng)絡(luò)的參數(shù)；重復(fù)上述步驟直至滿足停止訓(xùn)練的條件。

為解決上述技術(shù)問(wèn)題，本申請(qǐng)采用的另一個(gè)技術(shù)方案是：提供一種目標(biāo)重識(shí)別方法，該方法包括：獲取待識(shí)別圖像和參考圖像，其中，參考圖像包括預(yù)設(shè)目標(biāo)；利用目標(biāo)重識(shí)別網(wǎng)絡(luò)分別對(duì)待識(shí)別圖像和參考圖像進(jìn)行識(shí)別，以得到待識(shí)別圖像的特征和參考圖像的特征；基于待識(shí)別圖像的特征和參考圖像的特征之間的相似度，確定待識(shí)別圖像是否包括預(yù)設(shè)目標(biāo)；其中，目標(biāo)重識(shí)別網(wǎng)絡(luò)是利用上述方法得到的。

為解決上述技術(shù)問(wèn)題，本申請(qǐng)采用的又一個(gè)技術(shù)方案是：提供一種電子設(shè)備，該電子設(shè)備包括處理器、與處理器連接的存儲(chǔ)器，其中，存儲(chǔ)器存儲(chǔ)有程序指令；處理器用于執(zhí)行存儲(chǔ)器存儲(chǔ)的程序指令以實(shí)現(xiàn)上述方法。

為解決上述技術(shù)問(wèn)題，本申請(qǐng)采用的再一個(gè)技術(shù)方案是：提供一種存儲(chǔ)介質(zhì)，存儲(chǔ)有程序指令，該程序指令被執(zhí)行時(shí)能夠?qū)崿F(xiàn)上述方法。

通過(guò)上述方式，本申請(qǐng)訓(xùn)練圖像集中包括多個(gè)目標(biāo)對(duì)應(yīng)的多種風(fēng)格的圖像，因此利用目標(biāo)重識(shí)別網(wǎng)絡(luò)對(duì)訓(xùn)練圖像集中每張圖像識(shí)別，可以得到每個(gè)目標(biāo)對(duì)應(yīng)的多種風(fēng)格的圖像特征，進(jìn)而將每個(gè)目標(biāo)對(duì)應(yīng)的不同種風(fēng)格的圖像特征之間的距離(類內(nèi)距離)作為目標(biāo)重識(shí)別網(wǎng)絡(luò)的第一距離損失，并根據(jù)該第一距離損失調(diào)整目標(biāo)重識(shí)別網(wǎng)絡(luò)的參數(shù)，能夠使得目標(biāo)重識(shí)別網(wǎng)絡(luò)在后續(xù)識(shí)別過(guò)程中，得到的同一目標(biāo)對(duì)應(yīng)的不同風(fēng)格的圖像的特征之間距離更小，從而能夠減小圖像風(fēng)格差異(域間隙)對(duì)目標(biāo)重識(shí)別網(wǎng)絡(luò)識(shí)別結(jié)果帶來(lái)的影響，提高目標(biāo)重識(shí)別網(wǎng)絡(luò)的跨域識(shí)別性能，由于目標(biāo)重識(shí)別網(wǎng)絡(luò)能夠較好的跨域識(shí)別不同風(fēng)格圖像，故能夠提高目標(biāo)重識(shí)別網(wǎng)絡(luò)的目標(biāo)識(shí)別效果。

附圖說(shuō)明

圖1是本申請(qǐng)目標(biāo)重識(shí)別網(wǎng)絡(luò)的訓(xùn)練方法第一實(shí)施例的流程示意圖；

圖2是本申請(qǐng)目標(biāo)重識(shí)別網(wǎng)絡(luò)的訓(xùn)練方法第二實(shí)施例的流程示意圖；