---
library_name: litert
base_model: timm/visformer_small.in1k
tags:
  - vision
  - image-classification
datasets:
  - imagenet-1k
---

# visformer_small

Converted TIMM image classification model for LiteRT.

- Source architecture: visformer_small
- File: model.tflite

## Model Details

- **Model Type:** Image classification / feature backbone
- **Model Stats:**
  - Params (M): 40.2
  - GMACs: 4.9
  - Activations (M): 11.4
  - Image size: 224 x 224
- **Papers:**
  - Visformer: The Vision-friendly Transformer: https://arxiv.org/abs/2104.12533
- **Dataset:** ImageNet-1k
- **Original:** https://github.com/danczs/Visformer

## Citation

```bibtex
@inproceedings{chen2021visformer,
  title={Visformer: The vision-friendly transformer},
  author={Chen, Zhengsu and Xie, Lingxi and Niu, Jianwei and Liu, Xuefeng and Wei, Longhui and Tian, Qi},
  booktitle={Proceedings of the IEEE/CVF international conference on computer vision},
  pages={589--598},
  year={2021}
}
```