改進(jìn)YOLOv5 模型在自然環(huán)境下柑橘識(shí)別的應(yīng)用

打開(kāi)文本圖片集

摘要：在復(fù)雜的自然環(huán)境中綠色柑橘生長(zhǎng)形態(tài)各異，顏色與背景色相近，為有效識(shí)別綠色柑橘，提出一種基于混合注意力機(jī)制并改進(jìn)YOLOv5模型的柑橘識(shí)別方法。首先，改進(jìn)YOLOv5的網(wǎng)絡(luò)結(jié)構(gòu)，在主干網(wǎng)絡(luò)中添加混合注意力機(jī)制，即在主干網(wǎng)絡(luò)中的第2層嵌入SE（squeeze and excitation）注意力，第11層嵌入CA（coordinateattention）注意力；其次，改進(jìn)網(wǎng)絡(luò)模型特征融合結(jié)構(gòu)，將YOLOv5模型Concat特征融合操作的下層分支放在模型C3模塊之前，再與另一條上層分支進(jìn)行特征融合；最后，改進(jìn)模型分類(lèi)損失函數(shù)，將YOLOv5模型的分類(lèi)損失函數(shù)改成Varifocal Loss函數(shù)，加強(qiáng)綠色柑橘特征信息的提取，提高綠色柑橘檢測(cè)精度。（剩余13283字）