metadata
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
@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}
}