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

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