Plant Disease Classification with EfficientNetB4
Overview
This project leverages EfficientNetB4, a state-of-the-art convolutional neural network (CNN), to classify plant diseases using the PlantVillage dataset. The trained model achieves 97.66% accuracy on the test set, demonstrating robust performance in identifying crop diseases from leaf images.
Key Features
High Accuracy: 97.66% test accuracy, outperforming many traditional CNN architectures. Efficient Architecture: EfficientNetB4 balances computational efficiency and model capacity, making it suitable for deployment. Scalability: Adaptable to other plant disease datasets or extended to new disease classes.
Dataset
The model is trained on the PlantVillage dataset, which contains 54,305 images of healthy and diseased leaves across 14 crop species and 38 disease classes. Images are resized to 380x380 pixels to match EfficientNetB4’s input dimensions.
- Downloads last month
- 0
Model tree for liriope/PlantDiseaseDetection
Base model
google/efficientnet-b4