Research on grape leaf disease detection method based on NMA-YOLOv8n

School of Electronic and Information Engineering, Liaoning Technical University, Huludao, China

^* Corresponding author: ccp@lntu.edu.cn

Abstract

In response to the low inefficiency and high misjudgement rate of manually observing grape leaf diseases, an improved YOLOv8n grape leaf disease detection model NMA-YOLOv8n is proposed. Firstly, the global nonlinear attention NBL was introduced in the neck network, which enhances the backbone feature extraction capability by fusing the local and non-local attention, enabling the network to equally focus on small and normal targets. Secondly, the MPDCIoU loss function is designed to replace the original Bbox loss function, improving the regression accuracy of the bounding box. At the output end, the detection performance of the algorithm for small targets is improved by designing the AFPN small target detection head. The experimental results show that the NMA-YOLOv8n model mAP@0.5 reaches 94.7%, 1.7% higher than YOLOv8n; the FPS reaches 124.6 frames/sec, which can meet the real-time detection requirements, and has higher detection accuracy and speed compared with other five mainstream target detection models.NMA-YOLOv8n provides grape disease detection with a better method, which has certain significance for the prevention and control of grape diseases.