KMeans++與注意力機(jī)制融合的蘋(píng)果葉片病害識(shí)別方法
江蘇農(nóng)業(yè)科學(xué)
頁(yè)數(shù): 9 2024-10-20
摘要: 為解決復(fù)雜環(huán)境下小尺度蘋(píng)果葉片病害識(shí)別精度不高、魯棒性不強(qiáng)的問(wèn)題,在YOLO v5s的基礎(chǔ)上提出一種新的改進(jìn)方法。該方法首先在模型訓(xùn)練之前使用KMeans++聚類(lèi)算法生成更接近真實(shí)框的錨框;其次在骨干網(wǎng)絡(luò)中加入卷積塊注意幾模塊(convolutional block attention module, CBAM),來(lái)提升復(fù)雜環(huán)境下小目標(biāo)特征的提取能力;再次為了增強(qiáng)頸部網(wǎng)絡(luò)對(duì)不同... (共9頁(yè))