Apple Leaf Disease Detection Method Based on Improved YOLO v7

YUAN Jie, XIE Linwei, GUO Xu, LIANG Rongguang, ZHANG Yinggang, MA Haotian

Abstract

Apples have become one of the most popular fruits in the world, and the annual production of apples in China has continued to increase. However, there are certain diseases in the growth process of apple trees, which will affect the quality and yield of apples, resulting in economic losses of fruit farmers. Therefore, in view of the problem that apple leaf diseases have diverse forms and dense distribution, resulting in low detection accuracy, an improved YOLO v7 model was proposed to accurately detect apple leaf diseases. Firstly, bidirectional feature pyramid network (BiFPN) was used to replace the original feature fusion method in YOLO v7 to improve the model’s detection ability of different scale diseases on apple leaves. Secondly, after the ELAN and E-ELAN modules of YOLO v7, an efficient channel attention mechanism (ECA) was added to enhance the ability of the model to extract features of apple leaves disease and improve detection accuracy. Finally, the loss function of YOLO v7 was changed to the SIOU loss function to accelerate the convergence speed of the model. Experimental results showed that the improved YOLO v7 model had a precision of 89.4%, a recall rate of 81.5%, a mean average precision (mAP@0.5) of 90.5%, and a mean average precision (mAP@0.95) of 62.1%. Compared with the original YOLO v7 model, they were increased by 4.9, 5.2, 3.5, and 4.6 percentage points, respectively. Compared with the Faster R-CNN, SSD, YOLO v3, YOLO v5s, and YOLO v7 models, the mAP@0.5 of improved YOLO v7 model was increased by 40.9, 20.3, 4.0, 2.3 and 3.5 percentage points, respectively, and the single image detection speed reached 12ms. The research can provide a feasible technical means for accurately detecting apple leaf diseases.

 

Keyword: apple leaf ; disease detection ; YOLO v7 ; multi-scale fusion ; attention mechanism


Download Full Text:

PDF


References


TERENTEV A, DOLZHENKO V, FEDOTOV A, et al. Current state of hyperspectral remote sensing for early plant disease detection: a review[J] . Sensors, 2022, 22(3): 757.

XU W, ZHANG G, DUAN Y. Farmland detection in synthetic aperture radar images with texture signature J ]. Journal of Applied Remote Sensing, 2014, 8(1); 084997.

YANG X, YU Q, HE L, et al. The one-against-all partition based binary tree support vector machine algorithms for multi-class classification[ J ]. Neurocomputing, 2013, 113: 1 -7.

ZHANG D, LI M, XU D, et al. A survey on theory and algorithms for к-means problems [J]. Scientia Sinica Mathematica, 2020, 50(9) : 1387 - 1404.

WANG S, DU R, LIU Y. The learning and optimization of full bayes classifiers with continuous attributes [ J ]. Chinese Journal of Computers, 2012, 35(10): 2129 -2138.

CORCEIRO A, ALIBABAEI K, ASSUNCAO E, et al. Methods for detecting and classifying weeds diseases and fruits using Al to improve the sustainability of agricultural crops; a review [J] . Processes, 2023, 1 1(4) : 1263.

RAJPOOT V, TIWARI A, JALAL A S. Automatic early detection of rice leaf diseases using hybrid deep learning and machine learning methods[J]. Multimedia Tools and Applications, 2023, 82(23): 36091 -36117.

GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[ J]. arXiv Preprint, arXiv: 1311.2524, 2013.

GIRSHICK R. Fast R-CNN[J], arXiv Preprint, arXiv: 1504.08083, 2015.

REN S, HE K, GIRSHICK R, et al. Faster R — CNN: towards real-time object detection with region proposal networks [J] . IEEE Transactions on Pattern Analysis and Machine Intelligence, 2016, 39(6) : 1 137 - 1149.

GAO W Q, XIAO Z Y, BAO T F. Detection and identification of potato-typical diseases based on multidimensional fusion atrous-CNN and hyperspectral data [ J ]. Applied Sciences-Basel, 2023,13(8): 5023.

XUE Wei, CHENG Runhua, KANG Yalong, et al. Pear leaf disease spot counting method based on GC — Cascade R – CNN [J] . Transactions of the Chinese Society for Agricultural Machinery, 2022, 53(5) ; 237 -245. ( in Chinese)

ZHAOShengyi, LIU Jizhan, WU Shuo. Multiple disease detection method for greenhouse-cultivated strawberry based on multiscale feature fusion Faster R_CNN[J]. Computers and Electronics in Agriculture, 2022, 199 ; 107176.

LIU W, ANGUELOV D, ERHAN D, et al. SSD: single shot multibox detector [ J ]. arXiv Preprint, arXiv; 1512. 02325 , 2015.

CHITHAMBARATHANU M, JEYAKUMAR M K. Survey on crop pest detection using deep learning and machine learning approaches [ J ]. Multimedia Tools and Applications, 2023, 82(27) ; 42277 -42310.

TIAN L, ZHANG H, LIU B, et al. VMF — SSD; anovel V-space based multi-scale feature fusion SSD for apple leaf disease detection [J] . IEEE-ACM Transactions on Computational Biology and Bioinformatics, 2023, 20(3) : 2016 -2028.

ZHANG Lijie, ZHOU Shuhua, LI Na, et al. Apple location and classification based on improved SSD convolutional neural network [J] . Transactions of the Chinese Society for Agricultural Machinery, 2023, 54(6) : 223 -232. (in Chinese)

LEI Jianyun, YE Sha, XIA Meng, et al. Detection of grape leaf diseases based on improved YOLO [ J]. Journal of South- Central Minzu University ( Natural Science Edition) , 2022, 41(6) ; 712 -719. (in Chinese)

MATHEW M P, MAHESH T Y. Leaf-based disease detection in bell pepper plant using YOLO 5 [ J ]. Signal Image and Video Processing, 2022,16(3): 841 -847.

SUN Changlan, LIN Haifeng. An apple tree leaf disease detection method based on ensembel learingfj]. Jiangsu Agricultural Sciences, 2022, 50 ( 20) ; 41 - 47. (in Chinese)

ZHANG Lingxian, JING Jiaping, LI Shufei, et al. Tomato disease recognition system based on image automatic labeling and improved YOLO v5 [J]. Transactions of the Chinese Society for Agricultural Machinery, 2023, 54 ( 1 1 ); 198 - 207. (in Chinese)

W ANG Y, WANG Y, ZHAO J. MGA — YOLO: a lightweight one-stage network for apple leaf disease detection [J] . Frontiers in Plant Science, 2022, 13 : 927424.


Refbacks

  • There are currently no refbacks.