基于改进YOLOv8的行人摔倒检测算法
Pedestrian Fall Detection Algorithm Based on Improved YOLOv8
DOI: 10.12677/csa.2024.148173, PDF,   
作者: 王 震, 李 莉, 王 奇, 王树云:天津职业技术师范大学电子工程学院,天津
关键词: 行人摔倒YOLOv8注意力DiouInner-FocalerPedestrian Falling YOLOv8 Attention Diou Inner-Focaler
摘要: 随着社会老龄化进程的加快,行人摔倒事故成为了一个严重的社会问题。本文围绕基于改进YOLOv8的行人摔倒检测研究展开,针对原始YOLOv8在行人摔倒检测任务中存在的不足,提出了YOLOv8-RFAConv-Diou-Inner-Focaler模型。该模型通过RFA Conv卷积操作,在卷积层引入注意力机制,增强对重要特征的关注,提高特征的表达能力。引入Diou损失函数解决传统损失函数在目标重叠和尺度变化时的不足以及Inner-Focaler损失函数动态调整损失权重。通过实验验证:本文提出的改进算法在行人摔倒检测任务中取得了显著的性能提升,对比原始算法(YOLOv8算法)在平均精度上提高了4.8%。
Abstract: With the acceleration of social aging, pedestrian falling accidents have become a serious social problem. This article focuses on the research of pedestrian fall detection based on improved YOLOv8. In response to the shortcomings of the original YOLOv8 in pedestrian fall detection tasks, a YOLOv8-RFAConv-Diou-Inner-Focaler model is proposed. This model introduces attention mechanism in the convolutional layer through RFA Conv convolution operation, enhancing the attention to important features and improving the expression ability of features. Introducing the Diou loss function to address the shortcomings of traditional loss functions in target overlap and scale changes, as well as dynamically adjusting loss weights using the Inner Focaler loss function. Through experimental verification, the improved algorithm proposed in this paper has achieved significant performance improvement in pedestrian fall detection tasks, with an average accuracy improvement of 4.8% compared to the original algorithm (YOLOv8 algorithm).
文章引用:王震, 李莉, 王奇, 王树云. 基于改进YOLOv8的行人摔倒检测算法[J]. 计算机科学与应用, 2024, 14(8): 160-167. https://doi.org/10.12677/csa.2024.148173

参考文献

[1] 中华人民共和国国家统计局. 第七次全国人口普查公报解读[EB/OL]. 2021-05-11.
https://www.stats.gov.cn/sj/sjjd/202302/t20230202_1896484.html, 2021-08-30.
[2] 滕卓艳, 曹瑶瑶, 潘晓东, 等. 中国社区高龄老年人跌倒影响因素分析[J]. 伤害医学(电子版), 2023, 12(4): 17-23.
[3] 杨志勇, 王俊杰, 金磊. 基于SE-CNN的人体摔倒检测方法[J].计算机工程, 2022, 48(6): 270-277.
[4] 张靖. 改进YOLOv5的人体摔倒检测[J]. 现代信息科技, 2023, 7(4): 121-124.
[5] 李茁闻, 和晓军. 基于深度学习的人体摔倒检测算法设计[J]. 通信与信息技术, 2024(1): 101-105.
[6] 白亮, 丁学文, 申明坤, 等. 基于改进Yolov8的摔倒行为检测算法[J]. 天津职业技术师范大学学报, 2024, 34(1): 38-43.
[7] Bao, J., Li, S., Wang, G., Xiong, J. and Li, S. (2023) Improved YOLOV8 Network and Application in Safety Helmet Detection. Journal of Physics: Conference Series, 2632, Article ID: 012012. [Google Scholar] [CrossRef
[8] Zhu, J., Hu, T., Zheng, L., Zhou, N., Ge, H. and Hong, Z. (2024) YOLOv8-C2f-Faster-EMA: An Improved Underwater Trash Detection Model Based on YOLOv8. Sensors, 24, Article No. 2483. [Google Scholar] [CrossRef] [PubMed]
[9] 于春和, 任海宁. 基于改进YOLOv5s的老人跌倒检测[J]. 电脑与信息技术, 2023, 31(6): 27-29.
[10] Soni, V., Yadav, H., Bijrothiya, S. and Semwal, V.B. (2024) CABMNet: An Adaptive Two-Stage Deep Learning Network for Optimized Spatial and Temporal Analysis in Fall Detection. Biomedical Signal Processing and Control, 96, Article ID: 106506. [Google Scholar] [CrossRef
[11] 韩晶, 王伟宇, 吕学强, 等. 基于协同注意力与特征融合的无人机小目标检测[J]. 北京信息科技大学学报(自然科学版), 2023, 38(3): 1-8.
[12] Zhang, X., Liu, C., Yang, D., et al. (2023) RFAConv: Innovating Spatial Attention and Standard Convolutional Operation. arXiv: 2304.03198.
[13] Zheng, Z., Wang, P., Liu, W., Li, J., Ye, R. and Ren, D. (2020) Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression. Proceedings of the AAAI Conference on Artificial Intelligence, 34, 12993-13000. [Google Scholar] [CrossRef
[14] Zhang, H., Xu, C. and Zhang, S. (2023) Inner-IoU: More Effective Intersection over Union Loss with Auxiliary Bounding Box. arXiv:2311.02877.
[15] Lin, T., Goyal, P., Girshick, R., He, K. and Dollar, P. (2017). Focal Loss for Dense Object Detection. 2017 IEEE International Conference on Computer Vision (ICCV), Venice, 22-29 October 2017, 2999-3007.[CrossRef