uisikdag/autotrain-weeddatabal-vit-base-patch16-224

Aim

Classifies Plant Seedling Images (12 Class)

Trained on https://www.kaggle.com/datasets/vbookshelf/v2-plant-seedlings-dataset

Inference

from PIL import Image
#import requests

#url = 'https://..../forest.png'         # Can be any image from URL
#image = Image.open(requests.get(url, stream=True).raw)

# File Path
file_path = r"C:\......\1_resized.png"     # Can be any image from your storage

# Show image
image = Image.open(file_path)
#image.show()

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-classification", model="uisikdag/autotrain-weeddatabal-vit-base-patch16-224")
output=pipe(image)

Model Trained Using AutoTrain

• Problem type: Image Classification

Validation Metrics

loss: 0.10145530104637146

f1_macro: 0.9766586634653862

f1_micro: 0.9766666666666667

f1_weighted: 0.9766586634653862

precision_macro: 0.9769017094017093

precision_micro: 0.9766666666666667

precision_weighted: 0.9769017094017094

recall_macro: 0.9766666666666666

recall_micro: 0.9766666666666667

recall_weighted: 0.9766666666666667

accuracy: 0.9766666666666667