# PotatoLeafNet: two-stage convolutional neural networks for effective Potato Leaf disease identification and classification

## TL;DR

PotatoLeafNet is a deep learning model that accurately identifies potato leaf diseases using a two-stage approach, outperforming other models in accuracy and efficiency.

## Contribution

The novel two-stage framework combines fixed-sequence image augmentation with a compact CNN for efficient and accurate potato leaf disease classification.

## Key findings

- PotatoLeafNet achieves 98.52% accuracy on the PlantVillage-Potato test set, outperforming baseline models.
- The model demonstrates strong recall for disease classes and efficient learning with 10-epoch runs.
- The approach improves generalization and reduces computational cost compared to deeper or fused architectures.

## Abstract

Potato foliar diseases, particularly early and late blight, pose a serious threat to yield and food security, yet reliable visual recognition remains challenging due to cultivar heterogeneity, variable symptom expression, and acquisition noise in field-like imagery. To address these issues, we propose PotatoLeafNet, a two-stage deep learning framework that combines a fixed-sequence image-augmentation pipeline with a compact, task-optimized 11-layer convolutional neural network (CNN) using 3 × 3 kernels for robust, data-efficient classification of potato leaf conditions (Healthy, Early Blight, Late Blight).

We construct a dataset of 4,072 labeled potato leaf images from the PlantVillage-Potato subset and standardize all inputs to 224 × 224 RGB tensors with pixel intensities normalized to [0,1]. A balanced, fixed-order augmentation policy—comprising rotation, translation, shear, zoom, horizontal flipping, brightness adjustment, and channel jitter—is applied exclusively to the training split, increasing it to 6,000 images (2,000 per class) while keeping the validation and test sets free of synthetic samples. The second stage consists of an 11-layer CNN implemented in TensorFlow/Keras and trained with categorical cross-entropy loss and the Adam optimizer under a unified training and evaluation protocol. Performance is benchmarked against strong CNN and hybrid baselines, including ResNet-50 + VGG-16, VGG-16 + MobileNetV2, MobileNetV2, and Inception-V3.

On the PlantVillage-Potato test set, PotatoLeafNet achieves 98.52% accuracy, 98.67% macro-precision, 99.67% macro-recall, 99.16% macro-F1, and 1.00 macro-AUC, outperforming all baseline models under identical preprocessing and training conditions. In particular, PotatoLeafNet surpasses ResNet-50 + VGG-16 (97.10% accuracy, AUC 0.98), VGG-16 + MobileNetV2 (94.80% accuracy, AUC 0.93), MobileNetV2 (93.20% accuracy, AUC 0.92), and Inception-V3 (92.50% accuracy, AUC 0.91). Short 10-epoch runs yield stable convergence (training accuracy 88.22%, validation accuracy 86.91%, test accuracy 88.15%), indicating efficient learning from the augmented distribution.

The results demonstrate that explicitly coupling a fixed sequential augmentation stage with a lightweight 3×3-kernel CNN enables high tri-class accuracy, strong recall for disease classes, and improved generalization relative to deeper or fused architectures, without incurring substantial computational cost. By emphasizing disease-relevant structure while limiting overfitting, PotatoLeafNet provides a practical and resource-efficient solution for automated screening of potato leaf health in real-world agronomic settings, supporting timely and data-driven disease management.