Downloads · 30 days
10
1% of all-time downloads
bigcatvit
therealcyberlord/bigcatvit
bigcatvit is a image classification model from therealcyberlord. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Fine-tuning a Vision Transformer on the Big Cats Dataset In this project, we fine-tuned a vision transformer on the Big Cats dataset to perform image classification. The Big Cats dataset consists of 2339 images of 10…
All-time downloads
762
Public
Parameters
85.8M
687 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin343 MB · 50%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Mar 26, 2023
- Updated
- Mar 27, 2023
- SHA
- 0507423f
Model card
From the Hugging Face model README
Fine-tuning a Vision Transformer on the Big Cats Dataset In this project, we fine-tuned a vision transformer on the Big Cats dataset to perform image classification. The Big Cats dataset consists of 2339 images of 10 different types of big cats, including lions, tigers, jaguars, and more.
Our goal was to train a model that could accurately classify these images with high accuracy. After fine-tuning a pre-trained Vision Transformer, we were able to achieve an accuracy of 98%.
Kaggle dataset: https://www.kaggle.com/datasets/gpiosenka/cats-in-the-wild-image-classification
<img src="https://images.pexels.com/photos/8521833/pexels-photo-8521833.jpeg?auto=compress&cs=tinysrgb&w=1260&h=750&dpr=1">- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Mar 26, 2023
- Updated
- Mar 27, 2023