一种交通标志识别方法与系统

本发明涉及一种交通标志识别方法与系统，其中该方法，包括：获取交通标志图片；对所述交通标志图片进行数据增强形成增强后的交通标志图片；对所述增强后的交通标志图片进行标注形成训练样本；利用所述训练样本对改进的YOLOv5网络中进行训练得到交通标志识别模型；其中，改进的YOLOv5网络为添加了注意力机制的YOLOv5网络；利用所述交通标志识别模型检测目标交通标志图片得到识别结果。本发明通过在YOLOv5网络中添加注意力机制，可以帮助YOLOv5网络更好地捕捉图像中重要的特征信息，使得YOLOv5网络具有更好的适应性和泛化能力，大大提升了模型的检测精度。