融合注意力機制和上下文信息的實時交通標志檢測算法

摘要： 近年來交通標志檢測任務得到了廣泛的關注，然而現(xiàn)有方法常無法滿足檢測實時性要求，并且現(xiàn)有方法在檢測小尺度交通標志時存在較多漏檢情況，為此，提出融合注意力機制和上下文信息的實時交通標志檢測算法。該算法以YOLOv5為基準模型，首先，在主干網(wǎng)絡中嵌入空間注意力機制自適應地強化關鍵位置的特征，抑制干擾信息，提高主干網(wǎng)絡的特征提取能力；其次，設計跨階段局部窗口Transformer模塊用...