基于改進(jìn)YOLOv5的交通標(biāo)志識別方法
光電工程
頁數(shù): 14 2024-06-25
摘要: 交通標(biāo)志檢測是自動駕駛領(lǐng)域重要的環(huán)節(jié),針對當(dāng)前交通標(biāo)志的識別存在漏檢、誤檢、模型參數(shù)多,以及常見且復(fù)雜的代表性真實環(huán)境情況,如霧天魯棒性差的問題,提出一種改進(jìn)YOLOv5的小目標(biāo)交通標(biāo)志識別算法。首先對數(shù)據(jù)集進(jìn)行霧化操作以適應(yīng)在霧天情況下的準(zhǔn)確識別,使用更加輕量的部分卷積(partial convolution, PConv)構(gòu)建PC3特征提取模塊;隨后在頸部網(wǎng)絡(luò)中提出延伸的特...