Qiu J, Lu X, Wang X, Chen C, Chen Y, Yang Y. Research on image recognition of tomato leaf diseases based on improved AlexNet model.
Heliyon 2024;
10:e33555. [PMID:
39044970 PMCID:
PMC11263660 DOI:
10.1016/j.heliyon.2024.e33555]
[Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 01/20/2024] [Revised: 06/24/2024] [Accepted: 06/24/2024] [Indexed: 07/25/2024] Open
Abstract
Aiming at the problems that the traditional image recognition technology is challenging to extract useful features and the recognition time is extended; the AlexNet model is improved to improve the effect of image classification and recognition. This study focuses on 8 types of tomato leaf diseases and healthy leaves. By using HOG and LBP weighted fusion to extract image features, a tomato leaf disease recognition model based on the AlexNet model is proposed, and transfer learning is used to train the AlexNet model. Transfer the knowledge learned by the AlexNet model on the PlantVillage image dataset to this model while reducing the number of fully connected layers. Keras deep learning framework and programming language Python were used. The model was implemented, and the classification and identification of tomato leaf diseases were carried out. The recognition rate of feature-weighted fusion classification is higher than that of serial and parallel methods, and the recognition time is the shortest. When the weight coefficient ratio of HOG and LBP is 3:7, the image recognition rate is the highest, and its value is 97.2 %. From the model performance curve See, when the number of iterations is more than 150 times, the training set and test accuracy rate both exceed 97 %, the loss rate shows a gradient decline, and the change is relatively stable; compared with the traditional AlexNet model, HOG + LBP + SVM model, and VGG model, improved AlexNet model has the highest recognition rate, and it has high recall value, accuracy, and F1 value; Compared with the latest convolutional neural network disease recognition models, improved AlexNet model recognition accuracy was 98.83 %, and the F1 value was 0.994. It shows that the model has good convergence performance, fast prediction speed, and low loss rate and can effectively identify 8 types of tomato leaf images, which provides a reference for the research on crop disease identification.
Collapse