关键词:
交通标志检测
YOLO11
CAFM注意力机制
GAM注意力机制
摘要:
针对现有目标检测算法存在对交通标志识别准确率低、检测不完全等问题,提出一种将注意力机制融入YOLO11n的交通标志检测算法。首先,将卷积和注意力融合模块与YOLO11n主干部分结合,通过融合卷积操作和注意力机制,对图像的全局和局部特征进行有效建模,以提升检测精度;其次,利用全局注意力机制模块融入YOLO11n颈部,使模型对特征中的语义信息和位置信息提取更为充分,进而提高了模型的特征表达能力;最后,添加一个小目标检测层,保留更多浅层细节信息,以增强深层和浅层语义信息的融合,从而改善对小目标检测不完全的问题。实验结果证明,改进后的算法在TT100K数据集中,精确率、召回率、平均精度分别达83.9%,70.7%,82.4%,与原模型YOLO11n相比较,分别提高了5.7%,2.7%,6.3%,证明了改进算法的有效性。