基于改進YOLOv5算法的瓜蔞分級方法
中國農機化學報
頁數(shù): 8 2024-04-15
摘要: 為解決瓜蔞檢測技術存在的檢測精度低且檢測時間長的問題,提出一種基于改進YOLOv5算法的瓜蔞分級方法YOLOv5-GCB。在主干網(wǎng)絡引入Ghost卷積模塊替換傳統(tǒng)卷積,在保證準確率的同時減少模型的參數(shù)量;在特征提取網(wǎng)絡和推理層之間添加CA注意力模塊,增強模型對空間和通道信息的關注,提高檢測精度;在頸部網(wǎng)絡中引入雙向加權特征金字塔網(wǎng)絡(Bi-directional Feature... (共8頁)