therealcyberlord
commited on
Commit
•
4702b05
1
Parent(s):
eb30004
Update README.md
Browse files
README.md
CHANGED
@@ -8,5 +8,4 @@ In this project, we fine-tuned a vision transformer on the Big Cats dataset to p
|
|
8 |
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%.
|
9 |
|
10 |
|
11 |
-
|
12 |
-
[1] Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., ... & Houlsby, N. (2021). An image is worth 16x16 words: Transformers for image recognition at scale. arXiv preprint arXiv:2010.11929.
|
|
|
8 |
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%.
|
9 |
|
10 |
|
11 |
+
Kaggle dataset: https://www.kaggle.com/datasets/gpiosenka/cats-in-the-wild-image-classification
|
|