基于改進MobileViT的葡萄葉部病害識別模型
打開文本圖片集
摘要:本研究提出了一種優(yōu)化的葡萄葉部病害識別模型CD-MobileViT。首先,將MobileViT作為基礎(chǔ)網(wǎng)絡(luò),在Layer1、Layer2后面均嵌入坐標注意力模塊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 with Weight Decay Regularization),更好地控制模型復(fù)雜度并提高泛化能力。(剩余7179字)