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