基于改进YOLOv5的苹果叶病害识别算法

APPLE LEAF DISEASE IDENTIFICATION ALGORITHM BASED ON IMPROVED YOLOV5

  • 摘要: 针对苹果叶图像中病害面积小和背景环境复杂带来苹果叶病害识别精度低的问题,提出一种改进YOLOv5 的苹果叶病害检测方法。该方法利用 K-means 算法重新计算锚框,加快网络收敛速度;引入坐标注意力机制,增强复杂背景中病害的特征表达;通过 BiFPN 特征融合网络,增强网络多尺度特征聚合能力;使用 EIoU 作为 目标框的损失函数,提高回归精度。实验结果表明,改进方法的mAP值相比 YOLOv5 提高了1.4 百分点,并且与不同目标检测算法相比检测精度有明显提高。该识别算法可部署于移动端,为苹果病害的早期防治提供技术指导。

     

    Abstract: Aimed at the problem that the disease area in the apple leaf image is small and the background environment is complex, which brings about the low accuracy of apple leaf disease identification, a method to improve the detection of apple leaf diseases by YOLOv5 is proposed. This method used the k-means algorithm to recalculate the anchor box to speed up network convergence. Coordinate attention mechanisms were introduced to enhance the characteristic expression of diseases in complex backgrounds. Through the BiFPN feature fusion network, the multi-scale feature aggregation ability of the network was enhanced. EIoU was used as the loss function of the target box to improve the regression accuracy. Experimental results show that the mAP value of the improved method is increased by 1. 4 percentage points compared with YOLOv5, and the detection accuracy is significantly improved compared with different object detection algorithms. The identification algorithm can be deployed on mobile to provide technical guidance for the early prevention and control of apple disease.

     

/

返回文章
返回