适合车载边缘计算的拥挤行人检测算法

来源：计算机工程与应用杂志2023年第4期北京时间：

作者：帅泽群, 李军, 张世义

单位：重庆交通大学 机电与车辆工程学院，重庆 400041

摘要：针对目前车载计算单元的计算资源和计算能力有限，不能运行网络层次较深的目标检测算法，设计了一种轻量化的网络模型用于对拥挤行人场景的检测，将Darknet53骨干网络替换为GhostNet，通过引入线性计算获得与普通卷积相似的特征图来减少计算资源消耗；引入空间金字塔池化模块实现多尺度融合，加强特征提取；提出使用更加高效的搜索机制改进卷积块注意力机制模块，联合分类网络AlexNet对自适应搜索广度[k]值进行选取，进一步提高网络性能；采用Grad-CAM算法将网络模型实现热力图可视化来对注意力机制进行分析；引入CIOU损失函数实现真实框和预测值在中心点上的拟合，以此来加速模型收敛和实现更加精确的定位。研究结果表明：改进后的网络在WiderPerson行人检测数据集上行人类别查准率达到75.35%，相比于改进前的模型在行人查准率和平均查准率上分别提高了5.76个百分点和3.28个百分点。在Visdrone数据集上，改进后的网络平均查准率达到35.6%，在基本接近于YOLOv3的基础上，每秒检测图片的数量可以达到60张，相较于传统的单阶段检测算法，检测速率最高提升了52.1%，能满足移动设备以及车载计算对实时检测速度和精度的要求。

关键词：行人检测,神经网络,注意力机制,轻量化