基于時空注意力的骨骼行為識別方法
基本信息
申請?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)用前景。 |
