一种基于改进YOLOv5s模型的交通标识检测方法

    公开(公告)号:CN118298404A

    公开(公告)日:2024-07-05

    申请号:CN202410462556.6

    申请日:2024-04-17

    Applicant: 无锡学院

    Abstract: 本发明公开了一种基于改进YOLOv5s模型的交通标识检测方法,包括以下步骤:步骤1:获取交通标识图像数据集,并将交通标识图像数据集分为训练集和测试集;步骤2:改进YOLOv5s模型:将Neck网络中C3模块的3×3卷积替换为多样性分支DBB模块,得到C3_DB模块,在C3_DB模块后引入Context Aggregation模块;步骤3:将获得的训练集输入搭建好的YOLOv5s模型进行训练;用EIOU_Loss作为损失函数,当YOLOv5s模型损失曲线趋近于0且无明显波动时,停止训练,得到训练权重w,否则继续训练;步骤4:将获取到的交通标识图像输入到训练后的YOLOv5s模型中,进行交通标识检测,YOLOv5s模型根据权重自动识别交通标识的种类及数量。本发明改进后的YOLOv5s模型有效地提升了交通标志检测精度。

Patent Agency Ranking