基于改進(jìn)YOLOv5s的復(fù)雜環(huán)境下新梅檢測(cè)方法
農(nóng)業(yè)工程學(xué)報(bào)
頁(yè)數(shù): 8 2024-07-13
摘要: 為解決新梅在樹(shù)干樹(shù)葉遮擋、果實(shí)重疊情況下難以準(zhǔn)確檢測(cè)的問(wèn)題,該研究建立了新梅目標(biāo)檢測(cè)模型SFFYOLOv5s。在真實(shí)果園環(huán)境下構(gòu)建新梅數(shù)據(jù)集,以YOLOv5s模型作為基礎(chǔ)網(wǎng)絡(luò),首先在Backbone骨干網(wǎng)絡(luò)C3模塊中引入CA(coordinate attention)注意力機(jī)制以增強(qiáng)模型對(duì)新梅關(guān)鍵特征信息的提取能力并減少模型的參數(shù)量;其次在Neck層中引入加權(quán)雙向特征金字塔網(wǎng)絡(luò)...