关键词:
智能交通
车辆检测
轻量化
YOLOv7-tiny
深度学习
摘要:
在复杂道路环境下进行多目标检测时,现有算法存在识别效果差、参数量大、计算复杂度高等问题,不适合在资源有限的移动端设备部署。对此,基于YOLOv7-tiny提出一种结合非相邻特征的轻量化道路多目标检测算法。首先,设计轻量级非相邻特征金字塔网络Tiny-AFPN对不同尺度的非相邻特征进行融合,减少因尺度差异造成的特征损失,实现更丰富的跨尺度信息交互,提升模型的泛化能力。其次,通过引入分布偏移卷积DSConv,重新设计了高效层聚合网络ELAN,并命名为ELAN-DS,在优化高效层聚合网络、降低模型复杂度的同时提升特征表达能力。最后,采用MPDIoU损失函数,增强网络对目标的检测能力,提高了边界框回归的准确性。在SODA10M的实验中,改进后的YOLOv7-tiny算法相较于原模型的准确率、mAP@0.5和召回率分别提升了1.4%、1.4%、5.9%,参数量和计算量分别减少了8.2%和41.5%,有效降低了模型的参数量和计算复杂度,大幅度提升了模型的检测速度,为在边缘设备上的部署提供了可能性。