Image Classification
Transformers
Safetensors
English
vit
vision
vision-transformer
agriculture
plant-pathology
crop-disease-detection
Eval Results (legacy)
Instructions to use Armaan457/AgriVerse-Vision with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Armaan457/AgriVerse-Vision with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="Armaan457/AgriVerse-Vision") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("Armaan457/AgriVerse-Vision") model = AutoModelForImageClassification.from_pretrained("Armaan457/AgriVerse-Vision", device_map="auto") - Notebooks
- Google Colab
- Kaggle
AgriVerse Vision
AgriVerse-Vision is a vision model fine-tuned from google/vit-base-patch16-224 to classify crop health and diagnose 60 plant diseases across 16 crop species.
The model is the primary visual diagnostic component of the AgriVerse agricultural decision-support system.
Performance Highlights
- Test Accuracy: 98.91%
- Test F1-Score (Weighted): 98.90%
- Macro Average Precision / Recall / F1: 0.99 / 0.99 / 0.99
- Test Loss: 0.0483
- Evaluated on: 13,700 held-out test images
Model Overview
- Model Architecture: Vision Transformer (ViT-Base)
- Checkpoint Base:
google/vit-base-patch16-224 - Number of Parameters: ~86M
- Number of Classes: 60
- Input Resolution: 224 x 224 pixels (RGB)
Usage
- Foliar plant disease diagnosis and crop health classification
- Field scouting support and agricultural research
- Visual feature extractor for multimodal agricultural systems
Quickstart
Using transformers Pipeline
from PIL import Image
from transformers import pipeline
classifier = pipeline(
task="image-classification",
model="Armaan457/AgriVerse-Vision"
)
image = Image.open("leaf_image.jpg").convert("RGB")
predictions = classifier(image, top_k=5)
for pred in predictions:
print(f"{pred['label']:<32} Confidence: {pred['score']:.4f}")
- Downloads last month
- 1,495
Model tree for Armaan457/AgriVerse-Vision
Base model
google/vit-base-patch16-224Space using Armaan457/AgriVerse-Vision 1
Evaluation results
- Test Accuracy on Crop Disease and Plant Health Datasetself-reported0.989
- Test F1 (Weighted) on Crop Disease and Plant Health Datasetself-reported0.989
- Test Loss on Crop Disease and Plant Health Datasetself-reported0.048