关键词:
YOLOv5s
目标检测
AdamW优化器
SIoU损失函数
摘要:
近年来,人工智能技术的迅猛发展推动了自动驾驶技术的创新,为智能、高效、安全的出行方式提供了基础。然而,在面对识别小型物体时,尤其是在远距离或图像模糊的情况下,目标检测算法的性能和准确性受到了挑战。针对这一问题,本文提出了对YOLOv5s模型的优化方案。首先,引入了Soft-NMS机制取代传统的NMS,以减少冗余边界框的置信度分数。其次,采用SIoU损失函数替代CIoU损失函数,增强了模型的预测精度。最后,通过AdamW优化器对模型进行超参数优化,提高了训练效率。在MSCOCO128数据集上进行仿真验证,结果显示,改进后的算法mAP@0.5达94.9%,loss达0.0183,验证了模型的有效性和适用性。与其他模型相比,改进的YOLOv5s算法提高了4.4%的目标检测精度,尤其适用于解决自动驾驶中远处车辆过小和车辆重叠导致的识别困难。