Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
AlphaUncommon

kyegomez/VisionMamba

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

Owner
kyegomez (personal account)
Stars
Stars: 497
Forks
Forks: 23
Language
Python
License
MIT
Created
2024
Topics
aimachine-learningmambapytorchrecurrent-neural-networkssm
Tier
Alpha (250 to 699 ★) What this tier means
View on GitHub

More repos by kyegomez

All 40 repos by kyegomez

Nearby in Python

All Python repos