摘要:針對(duì)已有雜草識(shí)別模型對(duì)復(fù)雜農(nóng)田環(huán)境下多種目標(biāo)雜草的識(shí)別率低,、模型內(nèi)存占用量大,、參數(shù)多、識(shí)別速度慢等問(wèn)題,,提出了基于YOLO v5的輕量化雜草識(shí)別方法,。利用帶色彩恢復(fù)的多尺度視網(wǎng)膜(Multi-scale retinex with color restoration,MSRCR)增強(qiáng)算法對(duì)部分圖像數(shù)據(jù)進(jìn)行預(yù)處理,,提高邊緣細(xì)節(jié)模糊的圖像清晰度,,降低圖像中的陰影干擾。使用輕量級(jí)網(wǎng)絡(luò)PP-LCNet重置了識(shí)別模型中的特征提取網(wǎng)絡(luò),,減少模型參數(shù)量,。采用Ghost卷積模塊輕量化特征融合網(wǎng)絡(luò),進(jìn)一步降低計(jì)算量,。為了彌補(bǔ)輕量化造成的模型性能損耗,,在特征融合網(wǎng)絡(luò)末端添加基于標(biāo)準(zhǔn)化的注意力模塊(Normalization-based attention module,NAM),增強(qiáng)模型對(duì)雜草和玉米幼苗的特征提取能力,。此外,,通過(guò)優(yōu)化主干網(wǎng)絡(luò)注意力機(jī)制的激活函數(shù)來(lái)提高模型的非線性擬合能力。在自建數(shù)據(jù)集上進(jìn)行實(shí)驗(yàn),,實(shí)驗(yàn)結(jié)果顯示,,與當(dāng)前主流目標(biāo)檢測(cè)算法YOLO v5s以及成熟的輕量化目標(biāo)檢測(cè)算法MobileNet v3-YOLO v5s、ShuffleNet v2-YOLO v5s比較,,輕量化后雜草識(shí)別模型內(nèi)存占用量為6.23MB,,分別縮小54.5%,、12%和18%,;平均精度均值(Mean average precision,mAP)為97.8%,,分別提高1.3,、5.1、4.4個(gè)百分點(diǎn),。單幅圖像檢測(cè)時(shí)間為118.1ms,,達(dá)到了輕量化要求。在保持較高模型識(shí)別精度的同時(shí)大幅降低了模型復(fù)雜度,,可為采用資源有限的移動(dòng)端設(shè)備進(jìn)行農(nóng)田雜草識(shí)別提供技術(shù)支持,。