基于时空注意力的骨骼行为识别方法

基本信息

申请号 CN201910309644.1 申请日 -
公开(公告)号 CN110059620B 公开(公告)日 2021-09-03
申请公布号 CN110059620B 申请公布日 2021-09-03
分类号 G06K9/00 分类 计算;推算;计数;
发明人 汪从玲 申请(专利权)人 安徽省科亿信息科技有限公司
代理机构 合肥市上嘉专利代理事务所(普通合伙) 代理人 李璐;郭华俊
地址 230000 安徽省合肥市黄山路602号国家大学科技园创业孵化中心
法律状态 -

摘要

摘要 本发明公开了一种基于时空注意力的骨骼行为识别方法,包括以下步骤:S1:将获取的行为识别数据集进行预处理;S2:建立ST‑GCN网络和注意力单元,把注意力单元嵌入到ST‑GCN网络卷积层后面进行总体网络的搭建;S3:用训练集训练网络,优化参数,得到基于时空注意力的骨骼行为识别网络;S4:把测试集输入到步骤S3得到的网络进行预测,给出对应的动作类别。本发明提供一种带有注意力机制的图卷积行为识别方法,能够在图卷积行为识别方法的基础上进一步提高动作识别的准确率,在骨骼数据集上具有通用性,有广阔的应用前景。