Downloads · 30 days
0
animals10-resnet
MichaelMM2000/animals10-resnet
animals10-resnet is a image classification model from MichaelMM2000. Use it when you need a label for an image. It is set up for pytorch-lightning. The card lists the license as mit.
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.
Access
Public
Updated May 26, 2025
Repo size
44.9 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.ckpt44.9 MB · 100%
At a glance
- Task
- Image Classification
- Library
- pytorch-lightning
- License
- mit
- Access
- Public
- Created
- May 26, 2025
- Updated
- May 26, 2025
- SHA
- b30fa215
Datasets
Model card
From the Hugging Face model README
🐾 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.
🚀 Usage
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 consists of over 28,000 images across 10 animal categories.
📎 License
This model is released under the MIT License.
- Task
- Image Classification
- Library
- pytorch-lightning
- License
- mit
- Created
- May 26, 2025
- Updated
- May 26, 2025