Downloads · 30 days
6
21% of all-time downloads
henriquequeirozcunha/microvit-s2
microvit-s2 is a image classification model from henriquequeirozcunha. 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.
ImageNet-1K pretrained MicroViT-S2 (10.0M params, 74.6% Top-1 accuracy).
Downloads · 30 days
6
21% of all-time downloads
All-time downloads
28
Public
Parameters
11.5M
92.5 MB on disk
Likes
0
Public
Click a slice to open those files.
.pth46.3 MB · 50%
From the Hugging Face model README
ImageNet-1K pretrained MicroViT-S2 (10.0M params, 74.6% Top-1 accuracy).
Architecture: SHViT (Single-Head Vision Transformer) backbone. Source paper: https://arxiv.org/abs/2502.05800 Official repo: https://github.com/novendrastywn/MicroViT
from transformers import AutoModelForImageClassification, AutoImageProcessor
# Load ImageNet pretrained (1000 classes)
model = AutoModelForImageClassification.from_pretrained(
"henriquequeirozcunha/microvit-s2",
trust_remote_code=True,
)
# Fine-tune for binary classification
model = AutoModelForImageClassification.from_pretrained(
"henriquequeirozcunha/microvit-s2",
num_labels=2,
ignore_mismatched_sizes=True,
trust_remote_code=True,
)
Preprocessing: 224×224, ImageNet normalization (mean=[0.485,0.456,0.406], std=[0.229,0.224,0.225]).