《计算机应用》唯一官方网站 ›› 2023, Vol. 43 ›› Issue (6): 1811-1817.DOI: 10.11772/j.issn.1001-9081.2022050754

• 人工智能 • 上一篇    下一篇

复杂场景下的行人跌倒检测算法

方可, 刘蓉(), 魏驰宇, 张心月, 刘杨   

  1. 华中师范大学 物理科学与技术学院,武汉 430079
  • 收稿日期:2022-05-27 修回日期:2022-09-19 接受日期:2022-10-08 发布日期:2023-06-08 出版日期:2023-06-10
  • 通讯作者: 刘蓉
  • 作者简介:方可(1999—),男,河南周口人,硕士研究生,主要研究方向:深度学习、目标检测
    刘蓉(1969—),女,湖南安化人,副教授,博士,主要研究方向:智能信息处理、人工智能Email:liurong@ccnu.edu.cn
    魏驰宇(1998—),男,河南周口人,硕士研究生,主要研究方向:深度学习、目标检测
    张心月(1998—),女,河南周口人,硕士研究生,主要研究方向:智能信息处理、情感识别
    刘杨(1999—),男,湖南长沙人,硕士研究生,主要研究方向:智能信息处理、深度学习。
  • 基金资助:
    国家社会科学基金资助项目(19BTQ005)

Pedestrian fall detection algorithm in complex scenes

Ke FANG, Rong LIU(), Chiyu WEI, Xinyue ZHANG, Yang LIU   

  1. College of Physical Science and Technology,Central China Normal University,Wuhan Hubei 430079,China
  • Received:2022-05-27 Revised:2022-09-19 Accepted:2022-10-08 Online:2023-06-08 Published:2023-06-10
  • Contact: Rong LIU
  • About author:FANG Ke, born in 1999, M. S. candidate. His research interests include deep learning, object detection.
    WEI Chiyu, born in 1998, M. S. candidate. His research interests include deep learning, object detection.
    ZHANG Xinyue, born in 1997, M. S. candidate. Her research interests include intelligent information processing, sentiment identification.
    LIU Yang, born in 1999, M. S. candidate. His research interests include intelligent information processing, deep learning.
  • Supported by:
    National Social Science Foundation of China(19BTQ005)

摘要:

随着人口老龄化程度的不断深化,跌倒检测成为医疗与健康领域的一个关键问题。针对复杂场景下跌倒检测算法准确率偏低的问题,提出一种改进的跌倒检测模型——PDD-FCOS(PVT DRFPN DIoU-Fully Convolutional One-Stage object detection)。在基准FCOS算法的骨干网络中引入金字塔视觉转换器(PVT),以不增加计算量为前提提取更丰富的语义信息;在特征信息融合阶段插入双重细化特征金字塔网络(DRFPN),更加准确地学习特征图之间采样点的位置和其他信息,并通过上下文信息捕获特征通道之间更准确的语义关系,从而提升检测性能;训练阶段采用距离交并比(DIoU)损失进行边界框回归,通过优化预测框与目标框中心点的距离,使回归框收敛得更快更准确,从而有效提高跌倒检测算法的准确率。实验结果表明,所提模型在开源数据集Fall detection Database上平均精确度均值(mAP)达到82.2%,与基准FCOS算法相比,所提算法的mAP提升了6.4个百分点,且相较于其他主流目标检测算法有精度上的提升以及更好的泛化能力。

关键词: 目标检测, 行人跌倒检测, 金字塔视觉转换器, 注意力机制, 双重细化特征金字塔网络, 距离交并比

Abstract:

With the deepening of population aging, fall detection has become a key issue in the medical and health field. Concerning the low accuracy of fall detection algorithms in complex scenes, an improved fall detection model PDD-FCOS (PVT DRFPN DIoU-Fully Convolutional One-Stage object detection) was proposed. Pyramid Vision Transformer (PVT) was introduced into the backbone network of baseline FCOS algorithm to extract richer semantic information without increasing the amount of computation. In the feature information fusion stage, Double Refinement Feature Pyramid Networks (DRFPN) were inserted to learn the positions and other information of sampling points between feature maps more accurately, and more accurate semantic relationship between feature channels was captured by context information to improve the detection performance. In the training stage, the bounding box regression was carried out by the Distance Intersection Over Union (DIoU) loss. By optimizing the distance between the prediction box and the center point of the object box, the regression box was made to converge faster and more accurately, which improved the accuracy of the fall detection algorithm effectively. Experimental results show that on the open-source dataset Fall detection Database, the mean Average Precision (mAP) of the proposed model reaches 82.2%, which is improved by 6.4 percentage points compared with that of the baseline FCOS algorithm, and the proposed algorithm has accuracy improvement and better generalization ability compared with other state-of-the-art fall detection algorithms.

Key words: object detection, pedestrian fall detection, Pyramid Vision Transformer (PVT), attention mechanism, Double Refinement Feature Pyramid Networks (DRFPN), Distance Intersection over Union (DIoU)

中图分类号: