改進YOLOv5框架的交通標志檢測算法

摘要： 針對傳統(tǒng)YOLOv5算法識別交通標志精度低的問題，提出了一種改進的YOLOv5模型。首先，在骨干網絡中加入CA注意力機制，考慮了通道信息和方向相關的位置信息，提高檢測精度；然后，引入了DIoU＿NMS,它將中心點考慮進來保留更多的矩形框，提高遮掩重疊交通標志的識別精度；其次，加入了遞歸門控卷積，實現(xiàn)關鍵特征之間的高階交互；最后，在三個輸出檢測的基礎上增加了一個小目標檢測頭，增強... （共9頁）