基于YOLOv4改進(jìn)的交通標(biāo)志檢測算法
摘要: 自動駕駛中要盡早檢測到交通標(biāo)志及時作出行車決策,其中的交通標(biāo)志屬于小目標(biāo)。針對當(dāng)前小目標(biāo)檢測精度低的情況,提出基于YOLOv4改進(jìn)的交通標(biāo)志檢測算法。主要有以下幾部分:改進(jìn)骨干網(wǎng)絡(luò),嵌入融合注意力機(jī)制模塊,加強(qiáng)對通道和空間信息的關(guān)注;將交叉熵?fù)p失函數(shù)改為Focal Loss損失,解決樣本分布不平衡的問題;利用圖片的多尺度信息進(jìn)行特征提取,空洞卷積增加感受野。在TT00k數(shù)據(jù)集上... (共8頁)
開通會員,享受整站包年服務(wù)