光电子·激光2024,Vol.35Issue(3) :283-292.DOI:10.16136/j.joel.2024.03.0660

融合注意力机制的轻量级摔倒检测

Lightweight fall detection based on attention mechanism

李尧 李金哲 黄刚 周箩鱼
光电子·激光2024,Vol.35Issue(3) :283-292.DOI:10.16136/j.joel.2024.03.0660

融合注意力机制的轻量级摔倒检测

Lightweight fall detection based on attention mechanism

李尧 1李金哲 2黄刚 2周箩鱼1
扫码查看

作者信息

  • 1. 长江大学电子信息学院,湖北荆州 434023
  • 2. 国网湖北省电力有限公司松滋市供电公司,湖北荆州 434200
  • 折叠

摘要

摔倒检测大都依靠传感器设备,此类方法受设备自身和环境因素影响较大,常常无法发挥该有的作用,同时,基于视觉的方法往往实时性较差,鲁棒性不强.针对上述问题,本文提出了一种鲁棒性强、能有效部署在嵌入式设备上的轻量化摔倒检测算法.以YOLOv5为基准模型,首先,融合轻量级注意力机制模块,使网络更关注要识别的目标区域,增强网络的识别精度.其次,使用模型压缩方法对模型进行剪枝,减小模型体积和计算量,使模型轻量化,以提高推理速度和便于部署在嵌入式设备中.最后,对剪枝后的模型进行知识蒸馏,在不提升模型复杂度的前提下提升模型的检测精度.实验结果表明:本文模型相较于基准模型,mAP增加了 1.7%,召回率提高了 1.2%,模型体积减小了 79.1%,浮点运算量降低了 70.9%.将本文模型部署在嵌入式设备Jetson Nano上,检测速率达到13.2 frame/s,基本满足实时性摔倒检测的要求.

Abstract

Fall detection mostly depends on sensor equipment.The method is highly influenced by equip-ment and environmental factors,and often can not work well.In addition,vision-based methods are often not effective in terms of real-time and robust.In order to solve these problems,a lightweight fall detec-tion algorithm is proposed with strong robustness and convenient deployment in embedded devices.Tak-ing YOLOv5 as the benchmark model,the lightweight attention mechanism module is firstly integrated to make the network focus on the target area to be identified and enhance the recognition accuracy of the network.Secondly,the model is pruned by the model compression method,which reduces the volume and calculation.Therefore it makes the model lightweight,so as to improve the reasoning speed and facilitate deployment in embedded devices.Finally,knowledge distillation is carried out on the pruned model,which can improve the detection accuracy without increasing the complexity of the model.The experimental re-sults show that compared with the benchmark model,the mAP of this model is increased by 1.7%,the recall is increased by 1.2%,the model volume is reduced by 79.1%,and the floating-point operation is reduced by 70.9%.The proposed model is deployed on the embedded device Jetson Nano,and the detec-tion speed is up to 13.2 frame/s,which basically meets the requirements of real-time fall detection.

关键词

摔倒检测/注意力机制/轻量化/模型压缩/知识蒸馏

Key words

fall detection/attention mechanism/lightweight/model compression/knowledge distillation

引用本文复制引用

基金项目

国家自然科学基金(61901059)

国家自然科学基金(51978079)

出版年

2024
光电子·激光
天津理工大学 中国光学学会

光电子·激光

CSCD北大核心
影响因子:1.437
ISSN:1005-0086
参考文献量19
段落导航相关论文