Unofficial PyTorch implementation of MobileViT

Overview

This is a PyTorch implementation of MobileViT specified in “MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer“, arXiv 2021.

img

Usage

import torch
from mobilevit import mobilevit_xxs

net = mobilevit_xxs()
img = torch.randn(1, 3, 256, 256)
out = net(img)

Credits

Code adapted from MobileNetV2 and ViT.

GitHub

https://github.com/chinhsuanwu/mobilevit-pytorch

 

 

 

To finish reading, please visit source site