File size: 732 Bytes
17351b8 fe37668 eb30004 3ca7667 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
---
license: 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 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"> |