-
公开(公告)号:CN118587676A
公开(公告)日:2024-09-03
申请号:CN202410482624.5
申请日:2024-04-22
Applicant: 无锡学院
IPC: G06V20/58 , G06V10/25 , G06V10/44 , G06V10/80 , G06V10/82 , G06N3/045 , G06N3/0464 , G06N3/0495 , G06N3/082 , G06N3/084 , G06N3/0985
Abstract: 本发明公开了一种基于轻量化卷积神经网络的交通标志检测方法,通过使用Ghost卷积和深度卷积构建新的主干模块,成功减少了参数量和计算量,从而提高算法的效率;引入SPDC模块增强了算法对多尺度特征信息的提取能力,进一步提升了模型的检测准确性;引入CARAFE上采样算子有助于增强特征网络的空间分辨率,使模型更精确地还原目标的细节信息;通过采用Focal‑EIoU损失函数替换传统的CIoU损失函数,成功解决了类别不平衡和边界框回归问题,有效提高了检测精度;改进后的模型方法不仅有效提升了交通标志检测的准确性,而且易于部署到嵌入式设备中实现实时在线检测,进一步提升了自动驾驶环境感知的智能化水平。
-
公开(公告)号:CN118298404A
公开(公告)日:2024-07-05
申请号:CN202410462556.6
申请日:2024-04-17
Applicant: 无锡学院
IPC: G06V20/58 , G06V10/52 , G06V10/764 , G06V10/80 , G06V10/82 , G06N3/0464 , G06N3/048 , G06N3/08
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模型有效地提升了交通标志检测精度。
-