File size: 987 Bytes
b8dc1cc b3d291f b8dc1cc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | ---
license: apache-2.0
tags:
- image-classification
- vision-transformer
- microvit
library_name: transformers
---
# MicroViT-S2
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
## Usage
```python
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]).
|