计算机工程与科学

作者：王泽宇, 徐慧英, 朱信忠, 李琛, 刘子洋, 王子奕

单位：1.浙江师范大学计算机科学与技术学院,浙江 金华 321004;2.曼彻斯特大学理工学院计算机系,英国 曼彻斯特M139PL

关键词：目标检测,行人检测,轻量化,注意力机制,

基金：国家自然科学基金（62376252，61976196）；浙江省自然科学基金(LZ22F030003)；国家级大学生创新训练计划重点项目（202310345042）

在大型人员密集的场所中，人群异常聚集情况时有发生，对自动驾驶和大型公共场所人流量监控系统等应用场景中涉及到的密集行人检测技术带来了一定挑战，新一代的密集行人检测技术要求精确度更高、计算开销更小、检测速度更快以及部署更加方便等。针对上述需求，提出了一种基于YOLOv8改进的轻量级密集行人检测算法MER-YOLO，首先采用MobileViT作为主干网络，提升模型在总体上对行人聚集区域的特征提取能力；引入EMA注意力机制模块，对全局信息进行编码，通过维度交互来进一步聚合像素级特征，并结合160×160尺度的检测头加强小目标检测能力；使用排斥损失(Repulsion Loss)作为边界框损失函数，减少了人群密集情况下小目标行人的漏检误检的情况。实验结果表明，相较于YOLOv8n，MER-YOLO行人检测算法在Crowd Human数据集上mAP@0.5提升了4.5%，在WiderPerson数据集上mAP@0.5提升了2.1%，同时只有3.1×106的参数量和9.8 GFLOPs，满足低算力兼顾高精度的部署需求。

来源：2024年第6期

《计算机工程与科学》期刊编辑部

查看计算机工程与科学杂志2024年第6期