[發(fā)明專(zhuān)利]行人重識(shí)別方法及裝置

有效

• 申請(qǐng)?zhí)枺?/td>: 201810843200.1
• 申請(qǐng)日: 2018-07-27
• 公開(kāi)（公告）號(hào): CN109190470B
• 公開(kāi)（公告）日: 2022-09-27
• 發(fā)明（設(shè)計(jì)）人: 葛藝瀟;李卓婉;趙海宇;伊帥;王曉剛;李鴻升
• 申請(qǐng)（專(zhuān)利權(quán)）人: 北京市商湯科技開(kāi)發(fā)有限公司
• 主分類(lèi)號(hào): G06V40/10
• 分類(lèi)號(hào): G06V40/10;G06V10/74;G06V10/82;G06N3/04;G06N3/08
• 代理公司: 廣州三環(huán)專(zhuān)利商標(biāo)代理有限公司 44202
• 代理人: 郝傳鑫;熊永強(qiáng)

本發(fā)明實(shí)施例提供了行人重識(shí)別方法及裝置，包括：獲取包含目標(biāo)對(duì)象的查詢(xún)圖像和包括至少一張待搜索圖像的圖像集；利用預(yù)先訓(xùn)練的對(duì)抗生成網(wǎng)絡(luò)分別從查詢(xún)圖像和圖像集中提取目標(biāo)對(duì)象的身份特征和圖像集對(duì)應(yīng)的至少一組身份特征，其中，對(duì)抗生成網(wǎng)絡(luò)通過(guò)關(guān)鍵點(diǎn)信息和行人的身份表示信息訓(xùn)練得到；獲取目標(biāo)對(duì)象的身份特征與至少一組身份特征中每個(gè)身份特征的相似度；響應(yīng)于相似度符合預(yù)定條件的情況，從圖像集中獲取至少一張目標(biāo)圖像。本發(fā)明實(shí)施例，可以提高行人識(shí)別的準(zhǔn)確性。

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

本發(fā)明涉及計(jì)算機(jī)視覺(jué)技術(shù)領(lǐng)域，具體涉及行人重識(shí)別方法及裝置。

背景技術(shù)

為了保證人身安全、財(cái)產(chǎn)安全等，攝像頭監(jiān)控已成為監(jiān)控的一種重要方式，同時(shí)如何識(shí)別圖像中的人物也成為一個(gè)亟待解決的技術(shù)問(wèn)題。目前，一種常見(jiàn)的行人重識(shí)別方法為：提取圖像中人物的人物特征，根據(jù)人物特征識(shí)別人物。然而，由于圖像中人物姿態(tài)的多樣性，以致無(wú)法通過(guò)人物特征準(zhǔn)確地識(shí)別出行人。

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

本發(fā)明實(shí)施例提供行人重識(shí)別方法及裝置，用于提高行人識(shí)別的準(zhǔn)確性。

本發(fā)明實(shí)施例第一方面提供一種行人重識(shí)別方法，包括：

獲取包含目標(biāo)對(duì)象的查詢(xún)圖像和包括至少一張待搜索圖像的圖像集；

利用預(yù)先訓(xùn)練的對(duì)抗生成網(wǎng)絡(luò)分別從所述查詢(xún)圖像和所述圖像集中提取所述目標(biāo)對(duì)象的身份特征和所述圖像集對(duì)應(yīng)的至少一組身份特征，其中，所述對(duì)抗生成網(wǎng)絡(luò)通過(guò)關(guān)鍵點(diǎn)信息和行人的身份表示信息訓(xùn)練得到；

獲取所述目標(biāo)對(duì)象的身份特征與所述至少一組身份特征中每個(gè)身份特征的相似度；

響應(yīng)于所述相似度符合預(yù)定條件的情況，從所述圖像集中獲取至少一張目標(biāo)圖像。

在一個(gè)實(shí)施例中，所述對(duì)抗生成網(wǎng)絡(luò)包括圖像編碼器、生成器和辨別器，對(duì)抗生成網(wǎng)絡(luò)的訓(xùn)練過(guò)程包括：

獲取樣本行人圖像集、關(guān)鍵點(diǎn)信息和噪聲；

將所述樣本行人圖像集中任意兩張行人圖像輸入所述圖像編碼器，得到與所述兩張行人圖像對(duì)應(yīng)的兩個(gè)身份表示信息；

將所述兩個(gè)身份表示信息、所述關(guān)鍵點(diǎn)信息和所述噪聲輸入所述生成器，獲得兩張生成圖像；

根據(jù)所述兩張行人圖像、所述關(guān)鍵點(diǎn)信息、所述兩個(gè)身份表示信息、所述兩張生成圖像和所述辨別器，確定總損失；

根據(jù)所述總損失優(yōu)化所述圖像編碼器和所述生成器，獲得訓(xùn)練的對(duì)抗生成網(wǎng)絡(luò)。

在一個(gè)實(shí)施例中，所述辨別器包括驗(yàn)證辨別器、身份辨別器和姿態(tài)辨別器，所述根據(jù)所述兩張行人圖像、所述關(guān)鍵點(diǎn)信息、所述兩個(gè)身份表示信息、所述兩張生成圖像和所述辨別器，確定總損失包括：

將所述兩個(gè)身份表示信息輸入所述驗(yàn)證辨別器，確定第一損失；

將所述兩張行人圖像、所述兩張生成圖像和兩張真實(shí)圖像輸入所述身份辨別器，確定第二損失，所述真實(shí)圖像為所述生成圖像對(duì)應(yīng)的真實(shí)圖像；

將所述關(guān)鍵點(diǎn)信息、所述兩張生成圖像和所述兩張真實(shí)圖像輸入所述姿態(tài)辨別器，確定第三損失；

根據(jù)所述兩張生成圖像和所述兩張真實(shí)圖像，確定第四損失；

根據(jù)所述兩張生成圖像，確定第五損失；

將所述第一損失、所述第二損失、所述第三損失、所述第四損失和所述第五損失進(jìn)行加權(quán)疊加，獲得總損失。

在一個(gè)實(shí)施例中，所述將所述兩個(gè)身份表示信息輸入所述驗(yàn)證辨別器，確定第一損失包括：

獲取所述兩個(gè)身份表示信息之間的第一相似度；