融合注意力机制的轻量级摔倒检测
CSTR:
作者:
作者单位:

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

作者简介:

周箩鱼 (1985-),男,博士,副教授,硕士生导师,主要从事图像处理、人工智能方面的研究。

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(61901059,51978079)资助项目


Lightweight fall detection based on attention mechanism
Author:
Affiliation:

(1.Electronics and Information School, Yangtze University, Jingzhou, Hubei 434023, China;2.Songzi Power Supply Company, State Grid Hubei Electric Power Co., Ltd., Jingzhou, Hubei 434200, China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    摔倒检测大都依靠传感器设备,此类方法受设备自身和环境因素影响较大,常常无法发挥该有的作用,同时,基于视觉的方法往往实时性较差,鲁棒性不强。针对上述问题,本文提出了一种鲁棒性强、能有效部署在嵌入式设备上的轻量化摔倒检测算法。以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 equipment 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 detection algorithm is proposed with strong robustness and convenient deployment in embedded devices.Taking 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 results 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 detection speed is up to 13.2 frame/s,which basically meets the requirements of real-time fall detection.

    参考文献
    相似文献
    引证文献
引用本文

李尧,李金哲,黄刚,周箩鱼.融合注意力机制的轻量级摔倒检测[J].光电子激光,2024,35(3):283~292

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-09-24
  • 最后修改日期:2022-12-15
  • 录用日期:
  • 在线发布日期: 2024-02-26
  • 出版日期:
文章二维码