摘要
本发明公开了一种面向边缘设备的遮挡行人闯红灯姿态识别方法,本发明通过获取红灯亮起时的斑马线处监控行人图片,实现行人姿态关节点检测和闯红灯姿态识别两个任务。本发明通过学习构建姿态空间关系矩阵,实现对遮挡关节点特征的空间关系传播,修正错误的关节点定位。面向边缘设备计算能力有限的问题,本发明使用蒸馏学习思想,分布设计教师网络和学生网络,并设计了学生网络损失函数,来获得轻量级的姿态关节点定位的学生网络。最后针对闯红灯姿态识别任务,构建卷积神经网络,实现闯红灯和非闯红灯的状态识别。本发明具有场景适用能力强,关节点定位准确性高,轻量级网络,处理速度快,便于移植等优势,可有效实现行人闯红灯行为识别。
- 单位