摘要
针对自动驾驶的小目标车辆漏检问题,提出了一种基于YOLOv5s改进的车辆检测算法.该算法采用加权双向特征金字塔网络(BiFPN)的融合计算方式,可以加强不同层次的信息融合,同时保留更多的浅层语义信息.通过在主干网络中引入多头自注意力机制来提升网络特征提取能力.实验结果表明:改进的网络模型与原YOLOv5s 模型相比,均值平均精确率(mAP)提升了1.01%,其检测速度满足实时性需求,在不同光照条件下能够对小目标车辆进行有效的车辆目标检测.
基金项目
新能源汽车北京实验室建设项目(PXM2020_014224)