基于時空注意力的骨骼行為識別方法

基本信息

申請?zhí)?/td> CN201910309644.1 申請日 -
公開(公告)號 CN110059620B 公開(公告)日 2021-09-03
申請公布號 CN110059620B 申請公布日 2021-09-03
分類號 G06K9/00 分類 計算;推算;計數(shù);
發(fā)明人 汪從玲 申請(專利權(quán))人 安徽省科億信息科技有限公司
代理機構(gòu) 合肥市上嘉專利代理事務(wù)所(普通合伙) 代理人 李璐;郭華俊
地址 230000 安徽省合肥市黃山路602號國家大學(xué)科技園創(chuàng)業(yè)孵化中心
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于時空注意力的骨骼行為識別方法,包括以下步驟:S1:將獲取的行為識別數(shù)據(jù)集進行預(yù)處理;S2:建立ST?GCN網(wǎng)絡(luò)和注意力單元,把注意力單元嵌入到ST?GCN網(wǎng)絡(luò)卷積層后面進行總體網(wǎng)絡(luò)的搭建;S3:用訓(xùn)練集訓(xùn)練網(wǎng)絡(luò),優(yōu)化參數(shù),得到基于時空注意力的骨骼行為識別網(wǎng)絡(luò);S4:把測試集輸入到步驟S3得到的網(wǎng)絡(luò)進行預(yù)測,給出對應(yīng)的動作類別。本發(fā)明提供一種帶有注意力機制的圖卷積行為識別方法,能夠在圖卷積行為識別方法的基礎(chǔ)上進一步提高動作識別的準(zhǔn)確率,在骨骼數(shù)據(jù)集上具有通用性,有廣闊的應(yīng)用前景。