基于ViT-改進(jìn)YOLOv7的稻田雜草識別
農(nóng)業(yè)工程學(xué)報
頁數(shù): 9 2024-05-23
摘要: 為解決光線遮蔽、藻萍干擾以及稻葉尖形狀相似等復(fù)雜環(huán)境導(dǎo)致稻田雜草識別效果不理想問題,該研究提出一種基于組合深度學(xué)習(xí)的雜草識別方法。引入MSRCP(multi-scale retinex with color preservation)對圖像進(jìn)行增強(qiáng),以提高圖像亮度及對比度;加入ViT分類網(wǎng)絡(luò)去除干擾背景,以提高模型在復(fù)雜環(huán)境下對小目標(biāo)雜草的識別性能。在YOLOv7模型中主干特征... (共9頁)