GitHub repository: kyegomez/VisionMamba
Primary language: Python
Created: 2024-01-21T23:05:14Z
Description: 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