結(jié)合卷積和自注意力機(jī)制的交通標(biāo)志識(shí)別模型
計(jì)算機(jī)仿真
頁數(shù): 7 2023-09-15
摘要: 針對(duì)交通標(biāo)志識(shí)別精度低,識(shí)別模型參數(shù)量大,識(shí)別速度慢的問題,提出了將多頭自注意力機(jī)制應(yīng)用到Y(jié)OLOv4-tiny模型的主干網(wǎng)絡(luò)中,構(gòu)建D-CSP模塊和A-CSP模塊,替換原模型中的主干網(wǎng)絡(luò),其中D-CSP模塊使用深度可分離卷積顯著降低模型的參數(shù)數(shù)量和計(jì)算量,A-CSP模塊將多頭自注意力機(jī)制和卷積運(yùn)算融合,增強(qiáng)了特征圖的語義信息,最后在四層特征圖上進(jìn)行不同尺度的目標(biāo)檢測(cè),降低漏檢...