yolov论文-改进YOLOv5s的复杂交通场景路侧目标检测算法

针对传统路侧目标检测模型存在的对于行人、非机动车、受遮挡车辆等小目标检测精度低以及模型体积过大的问题，提出了一种基于改进 YOLOv5s 的路侧目标检测模型。首先，使用 EIoU Loss 替换原始的 CIoU Loss 作为目标边界框的回归损失函数，在加快预测框回归损失函数收敛速度同时提升了预测框的回归预测精度；其次，使用轻量级的通用上采样算子 CARAFE 替换原始的最近邻插值上采样模块，减少了上采样过程中特征信息的损失；然后，在原始的三尺度检测层的基础上新添加一层检测尺度更小的目标检测分支，并 提出了一种高效的解耦预测头对不同尺度的检测层进行解耦，进一步提升模型对于小目标的检测能力；最后，对改进后的模型进行通道剪枝，剪除对于检测效果影响不大的冗余通道，降低模型体积，使得模型更加适用于资源受限条件下的路侧目标检测任务。通过路侧目标检测数据集 DAIR-V2X-I 的实验结果表明，相较于原始 YOLOv5s 算法，改进后的算法在模型体积减小 5.7MB 的基础上，mAP50、mAP50:95 分别提高了 2.5%和3.8%，达到了 90.3%、67.7%，检测速度也明显提升