<p>This paper explores the use of optimized convolutional neural networks (CNNs) to classify diseases affecting potato leaves using TensorFlow-2. The dataset, sourced from Kaggle’s Plant Village repository, includes 152 images of healthy potato leaves and 1000 images each of early and late blight. The methodology covers data preparation, model architecture design, training, evaluation, and deployment. During data preparation, the data set was split into training sets (80%) and testing sets (20%), with images resized to 128x128 pixels. The Deep Learning (DL) models built using CNN with 4 different optimizers (ADAM, SGD, RMSPROP, and ADAMAX) and trained using a sparse categorical cross-entropy loss function, include multiple convolutional and pooling layers for feature extraction, and fully connected layers for classification. Early stopping was used to prevent overfitting. Model performance was assessed using accuracy, loss curves, confusion matrix, ROC curve, precision recall curve, classification report, and F1 score. In addition, we have used data augmentation to balance the dataset by increasing healthy potato leaves 6 times and the use of Ensemble Deep Learning (EDL). EDL10 which contains DL1 (CNN + ADAM), DL2 (CNN + SGD), DL3 (CNN + RMSPROP) and DL4 (CNN + ADAMX) performs best with a accuracy score of 97.0%. This highlights the importance of data balancing and the use of the ensemble classification approach for the detection of blight in Potato Leaves.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Optimized CNN-based ensemble deep learning approach for potato leaf disease detection with data augmentation

  • Achin Jain,
  • Arun Kumar Dubey,
  • Sunil K. Singh,
  • Arvind Panwar,
  • Neha Gupta,
  • Sudhakar Kumar,
  • Varsha Arya,
  • Wadee Alhalabi,
  • Shin-Hung Pan,
  • Bassma Saleh Alsulami,
  • Brij B. Gupta

摘要

This paper explores the use of optimized convolutional neural networks (CNNs) to classify diseases affecting potato leaves using TensorFlow-2. The dataset, sourced from Kaggle’s Plant Village repository, includes 152 images of healthy potato leaves and 1000 images each of early and late blight. The methodology covers data preparation, model architecture design, training, evaluation, and deployment. During data preparation, the data set was split into training sets (80%) and testing sets (20%), with images resized to 128x128 pixels. The Deep Learning (DL) models built using CNN with 4 different optimizers (ADAM, SGD, RMSPROP, and ADAMAX) and trained using a sparse categorical cross-entropy loss function, include multiple convolutional and pooling layers for feature extraction, and fully connected layers for classification. Early stopping was used to prevent overfitting. Model performance was assessed using accuracy, loss curves, confusion matrix, ROC curve, precision recall curve, classification report, and F1 score. In addition, we have used data augmentation to balance the dataset by increasing healthy potato leaves 6 times and the use of Ensemble Deep Learning (EDL). EDL10 which contains DL1 (CNN + ADAM), DL2 (CNN + SGD), DL3 (CNN + RMSPROP) and DL4 (CNN + ADAMX) performs best with a accuracy score of 97.0%. This highlights the importance of data balancing and the use of the ensemble classification approach for the detection of blight in Potato Leaves.