MohamedFiyaz
/

bird-species-classifier

Image Classification

Model card Files Files and versions

Bird Species Classifier (ViT)

Fine-tuned Vision Transformer (ViT-base-patch16-224) on the CUB-200-2011 dataset for 200-class bird species classification.

Model Details

Architecture: ViT-base-patch16-224 (Google)
Dataset: CUB-200-2011 (11,788 images, 200 species)
Training: Transfer learning with AdamW optimizer, 5 epochs
Test Accuracy: 99.93%
Framework: PyTorch + HuggingFace Transformers

Author

Mohamed Fiyaz

Downloads last month: 33

Safetensors

Model size

86M params

Tensor type

F32

·

Model tree for MohamedFiyaz/bird-species-classifier

Base model

google/vit-base-patch16-224

Finetuned

(2107)

this model

Space using MohamedFiyaz/bird-species-classifier 1