###
计算机系统应用英文版:2023,32(11):120-130
←前一篇   |   后一篇→
本文二维码信息
码上扫一扫!
基于MCA-YOLOv5s的轻量化地铁站内行人检测
(1.南京信息工程大学 电子与信息工程学院, 南京 210044;2.南京国电南自轨道交通工程有限公司, 南京 210032)
Lightweight Subway Pedestrian Detection Based on MCA-YOLOv5s
(1.School of Electronic and Information Engineering, Nanjing University of Information Science and Technology, Nanjing 210044, China;2.Nanjing Guodian Nanzi Rail Transit Engineering Co. Ltd., Nanjing 210032, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 478次   下载 1085
Received:April 18, 2023    Revised:May 17, 2023
中文摘要: 随着智慧车站和云计算的迅速发展, 地铁站内大规模视频监控系统行人检测的部署愈发重要, 在客流监测、乘客引导和行为警示等方面发挥着人力不能及的重要作用. 在实际工程应用中, 受到计算资源有限以及多尺度多角度遮挡的困难样本带来错漏检的不利影响, 为此提出一种轻量化行人检测算法MCA-YOLOv5s. 首先使用MobileNetv3代替YOLOv5主干网络, 实现网络模型轻量化处理, 并用PConv代替MobileNetv3网络中的DWConv, 减少冗余计算和内存访问; 其次在特征融合阶段的C3模块中融入坐标注意力机制, 使模型更加关注行人的位置信息; 同时将损失函数CIoU替换为Alpha IoU以增加High Loss目标的权重和边界框的回归精度; 最后通过FPGM剪枝压缩改进后的网络模型, 提升模型加载和运行速度. 将改进后的模型部署在华为Atlas 300 AI加速卡中, 对地铁站内行人进行检测, 其平均精度达到94.1%, 检测速度为104.1 fps. 实际工程实践表明, 改进后的算法检测速度提升71.8%, 节省了站内硬件部署资源, 更满足地铁大客流下的行人监测和管理的工程实际需求.
Abstract:With the rapid development of smart stations and cloud computing, the deployment of large-scale video surveillance systems for pedestrian detection in subway stations is becoming more and more important, which plays an important role in passenger flow monitoring, passenger guidance, and behavior warning. In practical engineering applications, a lightweight pedestrian detection algorithm MCA-YOLOv5s is proposed due to the adverse effects of limited computing resources and difficult samples caused by multi-scale and multi-angle occlusion. Firstly, MobileNetv3 replaces the YOLOv5 backbone network to achieve lightweight network model processing, and PConv replaces DWConv in the MobileNetv3 network to reduce redundant computation and memory access. Secondly, the coordinate attention mechanism is incorporated in the C3 module of the feature fusion stage to make the model pay more attention to pedestrian position information. At the same time, the loss function CIoU is replaced by Alpha IoU to increase the weight of the High Loss target and the regression accuracy of the bounding box. Finally, the improved network model is compressed by FPGM pruning to improve the loading and running speed of the model. The improved model is deployed in Huawei Atlas 300 AI accelerator to detect pedestrians in subway stations. The average accuracy is 94.1%, and the detection speed is 104.1 fps. The actual engineering practice shows that the detection speed of the improved algorithm is increased by 71.8%, saving the hardware deployment resources in the station and meeting the actual engineering needs of pedestrian monitoring and management in subway stations with large passenger flow.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金(U21B2003); 江苏省产业前瞻与关键核心技术竞争项目(BE2022075)
引用文本:
孙同庆,刘光杰,唐喆,李佑文.基于MCA-YOLOv5s的轻量化地铁站内行人检测.计算机系统应用,2023,32(11):120-130
SUN Tong-Qing,LIU Guang-Jie,TANG Zhe,LI You-Wen.Lightweight Subway Pedestrian Detection Based on MCA-YOLOv5s.COMPUTER SYSTEMS APPLICATIONS,2023,32(11):120-130