基于改进SSD的骑行人员佩戴头盔检测

摘要

为了解决骑行人员佩戴头盔检测任务中目标小、密集、准确率差、检测速度慢、应用困难等问题，本文基于SSD网络提出了EfficientNetV2-SSD算法。针对原SSD网络参数多的问题，使用改进后的轻量级网络EfficientNetV2替换SSD中的特征提取网络，减少网络参数，提升网络检测速度；针对难检测的小目标，使用自上而下与自下而上的FPN金字塔结构，最大程度丰富所有预测特征层信息，提升小目标的检测准确率；针对头盔等被检测的目标特征，重新设计先验框尺寸与比例，提高了小目标检测的准确率，同时加快网络收敛速度，减小网络体积。实验结果表示，EfficientNetV2-SSD网络对佩戴头盔的检测平均精度均值相比SSD网络提高7.01%,网络体积减少75%,具有更好的实用性。