Strawberry disease identification based on improved YOLOv5
In order to improve the total yield of strawberries,reasonable monitoring and control of strawberry diseases is an effective means,a strawberry disease identification algorithm based on improved YOLOv5 is proposed.The detection algorithm uses CSPDarknet as the backbone feature extraction network,which can effectively improve the performance and training efficiency of the model.The EIOU loss function and K-means clustering algorithm are used to improve the convergence speed of the model.At the same time,CBAM attention mechanism is added to the model to improve the detection accuracy,and finally the CBAM-YOLOv51 algorithm based on improved YOLOv5 is constructed.The experimental results show that the improved model improves the detection accuracy and still ensures efficient detection speed compared to the original model.In addition,the trained CBAM-YOLOv51 target detection algorithm achieves an overall average accuracy of 96.52%under the validation set,with an average detection time of 27.52 ms.Compared with YOLOv4,YOLOv4-Tiny,Faster_R-CNN and other target detection algorithms,CBAM-YOLOv51 algorithm has greater advantages in accuracy.It has good robustness and real-time performance in the actual strawberry orchard environment,and it can meet the needs of strawberry disease identification accuracy and reliably prompt the health status of strawberries,so as to timely achieve precise pesticide application and other protection measures.