File size: 1,342 Bytes
b30fa21 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
---
license: mit
tags:
- image-classification
- computer-vision
- animals10
- resnet
- pytorch-lightning
datasets:
- custom
library_name: pytorch-lightning
---
# 🐾 Animal Breed Classifier (ResNet18 - Animals10)
This model is a fine-tuned ResNet18 trained on the Animals-10 dataset using PyTorch Lightning. It classifies images into one of 10 animal categories.
## 🧠 Model Details
- Architecture: ResNet18
- Framework: PyTorch Lightning
- Training dataset: Animals-10
- Input size: 224x224 RGB
- Classes: ['butterfly', 'cat', 'chicken', 'cow', 'dog', 'elephant', 'horse', 'sheep', 'spider', 'squirrel']
## 📊 Performance
| Metric | Value |
|---------------|-------|
| Accuracy | 95% |
| Macro F1 | 94% |
| Weighted F1 | 95% |
For a detailed comparison with zero-shot CLIP, see the [GitHub repo](https://github.com/YOUR_USERNAME/YOUR_REPO).
## 🚀 Usage
```python
from models.lightning_model import Classifier
model = Classifier.load_from_checkpoint(
'https://huggingface.co/MichaelMM2000/animals10-resnet/resolve/main/resnet_animals10.ckpt',
num_classes=10
)
```
## 🧪 Dataset
The [Animals-10 dataset](https://www.kaggle.com/datasets/alessiocorrado99/animals10) consists of over 28,000 images across 10 animal categories.
## 📎 License
This model is released under the MIT License.
|