改進(jìn)YOLOv5框架的交通標(biāo)志檢測算法

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