基于改進(jìn)MobileViT的葡萄葉部病害識(shí)別模型
山東農(nóng)業(yè)科學(xué)
頁數(shù): 8 2024-09-10
摘要: 本研究提出了一種優(yōu)化的葡萄葉部病害識(shí)別模型CD-MobileViT。首先,將MobileViT作為基礎(chǔ)網(wǎng)絡(luò),在Layer1、Layer2后面均嵌入坐標(biāo)注意力模塊CA(Coordinate Attention),以使網(wǎng)絡(luò)能更有效地捕捉不同位置的關(guān)鍵特征;其次,在網(wǎng)絡(luò)全連接層之后添加Dropout層,防止數(shù)據(jù)出現(xiàn)過擬合現(xiàn)象;最后,選用結(jié)合權(quán)重衰減的優(yōu)化器AdamW(Adam wit... (共8頁)
開通會(huì)員,享受整站包年服務(wù)