giriprasad51 / image_classification_project

Fine-tuning Transformers For Image Classification

name	self - supervised	initialized checkpoint	crop size	#params	accuracy	Kaggle Notebook
ViT	[ ✗ ]	vit_checkpoint	224x224	85M	0.9561	cifar-10-mobilevit-s.ipynb
BEiT	[ ✓ ]	beit_checkpoint	224x224	87M	0.9505	cifar-10-by-fine-tuning-beit.ipynb
mobilevit_s	[ ✗ ]	model.pth	224x224	5M	0.8900	cifar-10-mobilevit-s.ipynb

Confusion Matrixs

ViT	BEiT	Vit-s

Setup

First, clone the repo and install required packages:

!git clone https://github.com/giriprasad51/image_classification_project.git
cd image_classification_project
pip install -r requirements.txt

Demo