改进YOLOv8的果蔬病害检测算法

IMPROVED YOLOV8 ALGORITHM FOR FRUIT AND VEGETABLE DISEASE DETECTION

  • 摘要: 为解决目前农业中果蔬的疾病识别技术精度低、容易漏检误检等问题,提出一种基于YOLOv8改进的YOLOv8-GFPN算法。使用GFPN网络替换原来YOLOv8的Neck网络,提高模型的特征提取能力;使用C2f-fast-EMA替换原来的C2f模块,减少模型参数量和计算量;引入Wise-IoU替代原来的CIoU损失函数,提高模型的整体性能。为了验证改进后的YOLOv8-GFPN算法先进性,将其与YOLOv8原模型在果蔬病害数据集上进行检测结果对比,mAP@0.5上升2.9百分点,mAP@0.5:0.95上升5.1百分点,GFLOPs比原来减少了17%。实验结果表明,改进后的算法更适用于果蔬病害识别。

     

    Abstract: To address the issues of low accuracy and missed or false detection technology for fruits and vegetables, an improved YOLOv8 algorithm named YOLOv8-GFPN is proposed. The GFPN network was used to replace the original YOLOv8 Neck network to enhance the model’s feature extraction ability. C2f-fast-EMA was employed to replace the original C2f module, reducing model parameters and computational complexity. Wise-IoU was introduced to replace the original CIoU loss function, improving overall model performance. To verify the advancement of improved YOLOv8-GFPN algorithm, the improved algorithm was compared with original YOLOv8 on fruit and vegetable disease dataset. mAP@0.5 increased by 2.9 percentage points, mAP@0.5:0.95 increased by 5.1 percentage points, GFLOPs reduced by 17%. Experimental results show that the improved algorithm is more suitable for fruit and vegetable disease recognition.

     

/

返回文章
返回