Plant Disease Classification Model
A PyTorch model for classifying plant diseases in Apple, Tomato, and Corn crops.
Model Details
- Model Type: Image Classification
- Architecture: EfficientNet-B2 with Attention
- Input Size: 224x224 RGB images
- Output: Disease classification
Usage
from inference import load_model, predict_image
# Load model
model, class_names = load_model("model.pth")
# Make prediction
results = predict_image("your_image.jpg", model, class_names)
print(results)