We found a match
Your institution may have access to this item. Find your institution then sign in to continue.
- Title
AN IMPROVED CNN MODEL FOR CLASSIFICATION OF APPLE LEAF DISEASE AND VISUALIZATION USING WEIGHTED GRADIENT CLASS ACTIVATION MAP.
- Authors
Mahato, Dharmendra Kumar; Pundir, Amit; Saxena, Geetika Jain
- Abstract
Convolutional Neural Network (CNN), a particular type of forwarding feed network composed of convolutional, pooling, and fully connected layers, has become the dominant and most widely used deep learning architecture. Significantly enhanced effectiveness of ConvNets has made CNNs the go-to architecture model for almost every image processing-based application. CNNs automatically and adaptively learn spatial hierarchies of features with high accuracy, precision, and efficiency. This paper proposes three CNN models with 5, 6, and 7 layers with two types of classification layers at the top of the model, resulting in six kinds of models. Each model is trained on apple leaf diseases obtained with augmentation deployed on the PlantVillage dataset containing images of healthy and three types of leaf diseases. The trained models are compared on training time, testing accuracy, testing time. The best performing model (6-layer based model with fully connected layer as a classifier (6FC) in our case) yields 99.14% accuracy. This best-performing model is also compared with the state-of-art models such as VGG-16, InceptionV3, and MobileNetV2, trained using the transfer learning approach. After model comparison, we found our best model (6FC) outperformed the other models based on evaluated performance metrics with improvements as 3.94% gain in accuracy, 25.97% reduced parameters, and less training time (0.51hr) and testing time (20.5 sec) compared to VGG-16. Comparing precision, recall, and f1-score values are also found high (between 0.98 to 1) with our proposed model. The weighted gradient class activation map (Grad-CAM) technique generates a visualization of class predictions on the test dataset. The Grad-Cam visualization of results validates the prediction score attained by the proposed model.
- Subjects
CONVOLUTIONAL neural networks; DEEP learning; VISUALIZATION
- Publication
ICTACT Journal on Image & Video Processing, 2022, Vol 12, Issue 3, p2615
- ISSN
0976-9099
- Publication type
Article
- DOI
10.21917/ijivp.2022.0371