面向拥挤行人检测的改进DETR算法

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

作者：樊嵘, 马小陆

单位：安徽工业大学 电气与信息工程学院，安徽 马鞍山 243002

摘要：拥挤行人检测是行人检测领域的研究热点。针对拥挤行人检测场景中被遮挡目标及小目标行人易产生漏检的问题，提出一种改进DETR目标检测算法。针对拥挤行人场景中遮挡目标特征缺失的问题，采用注意力模型DETR作为基准模型，使模型可以在缺失部分特征的前提下完成目标检测；针对DETR模型对小目标行人检测效果差的问题，引入可变形注意力编码器，使模型可以有效利用含有大量小目标信息的多尺度特征图提升对小目标行人的检测精度；针对ResNet-50骨干网络对重要特征提取及提纯效率较低的问题，采用融合了通道空间注意力模块的改进EfficientNet骨干网络作为特征提取网络，提升模型对重要特征的提取能力以及提纯效率；针对采用注意力检测模块的模型训练效率较低的问题，训练时将Smooth-L1与GIOU结合作为损失函数，使模型可以进一步收敛至更高精度。在Wider-Person拥挤行人检测数据集上的实验结果表明，所提算法领先YOLO-x算法0.039的AP50精度，领先YOLO-v5算法0.015的AP50精度。该算法可以较好地运用于拥挤行人检测任务。

关键词：机器视觉,拥挤行人检测,注意力机制,DETR算法