Add LiteRT converted swin_tiny_patch4_window7_224

546f23e verified 10 days ago

1.37 kB

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

swin_tiny_patch4_window7_224

Converted TIMM image classification model for LiteRT.

Source architecture: swin_tiny_patch4_window7_224
File: model.tflite

Model Details

Model Type: Image classification / feature backbone
Model Stats:
- Params (M): 28.3
- GMACs: 4.5
- Activations (M): 17.1
- Image size: 224 x 224
Papers:
- Swin Transformer: Hierarchical Vision Transformer using Shifted Windows: https://arxiv.org/abs/2103.14030
Original: https://github.com/microsoft/Swin-Transformer
Dataset: ImageNet-1k

Citation

@inproceedings{liu2021Swin,
  title={Swin Transformer: Hierarchical Vision Transformer using Shifted Windows},
  author={Liu, Ze and Lin, Yutong and Cao, Yue and Hu, Han and Wei, Yixuan and Zhang, Zheng and Lin, Stephen and Guo, Baining},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
  year={2021}
}

@misc{rw2019timm,
  author = {Ross Wightman},
  title = {PyTorch Image Models},
  year = {2019},
  publisher = {GitHub},
  journal = {GitHub repository},
  doi = {10.5281/zenodo.4414861},
  howpublished = {\url{https://github.com/huggingface/pytorch-image-models}}
}