---
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.