https://github.com/scott-yjyang/Vivim

Vivim: a Video Vision Mamba for Medical Video Segmentation
https://github.com/scott-yjyang/Vivim

Last synced: 3 months ago
JSON representation

Vivim: a Video Vision Mamba for Medical Video Segmentation

• Host: GitHub
• URL: https://github.com/scott-yjyang/Vivim
• Owner: scott-yjyang
• Created: 2024-01-24T11:59:07.000Z (12 months ago)
• Default Branch: main
• Last Pushed: 2024-10-24T10:51:18.000Z (3 months ago)
• Last Synced: 2024-10-25T04:30:44.074Z (3 months ago)
• Language: Python
• Homepage:
• Size: 9.29 MB
• Stars: 145
• Watchers: 4
• Forks: 6
• Open Issues: 10
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• Awesome-Mamba - Vivim: a Video Vision Mamba for Medical Video Object Segmentation
• Awesome-state-space-models - Video

README

        
# Vivim

Vivim: a Video Vision Mamba for Medical Video Segmentation 

[[arXiv](https://arxiv.org/abs/2401.14168)]

## News

- 24-08-01. Upload several example cases of VTUS dataset

- 24-03-11. ❗❗Update on Code. Welcome to taste.😄

- 24-02-08. Update on Method and Experiments.

- 24-01-26. This project is still quickly updating 🌝. Check TODO list to see what will be released next.

- 24-01-25. The paper has been released on arXiv.

## A Quick Overview 



## Environment Setup

### Clone this repository and navigate to the root directory of the project.

```bash

git clone https://github.com/scott-yjyang/Vivim.git

cd Vivim

```

### Install basic package

```bash

conda env create -f environment.yml

```

### Install casual-conv1d

```bash

cd causal-conv1d

python setup.py install

```

### Install mamba

```bash

cd mamba

python setup.py install

```

### TODO LIST

- [x] Release Model

- [x] Release training scripts

- [x] Release evaluation

- [ ] Release Ultrasound dataset

- [ ] Experiments on other video object segmentation datasets.

- [ ] configuration

## Thanks

Code is based on [hustvl/Vim](https://github.com/hustvl/Vim), [bowang-lab/U-Mamba](https://github.com/bowang-lab/U-Mamba).

## Cite

If you find it useful, please cite and star

~~~

@article{yang2024vivim,

  title={Vivim: a Video Vision Mamba for Medical Video Object Segmentation},

  author={Yang, Yijun and Xing, Zhaohu and Zhu, Lei},

  journal={arXiv preprint arXiv:2401.14168},

  year={2024}

}

~~~