Tutorial

Install Mamba via https://github.com/state-spaces/mamba

Framework

Result

Train

python ./scripts/torch/train_cross.py --gpu 1 --epochs 1 --batch-size 1 --model-dir output/train_debug --model mm-feat

Test

python ./scripts/torch/test_cross.py --gpu 0 --model mm-feat --load-model "/home/guotao/code/voxelmorph-dev/output/train_s46/min_train.pt"

Data

We implement MambaMorph on our brain MR-CT data SR-Reg which is developed from SynthRAD 2023 (https://synthrad2023.grand-challenge.org/). Please go Here to access the SR-Reg dataset.

Data sample

Paper

https://arxiv.org/abs/2401.13934

Citation

@article{wang2025mamba,

title={Mamba-based deformable medical image registration with an annotated brain MR-CT dataset},

author={Wang, Yinuo and Guo, Tao and Yuan, Weimin and Shu, Shihao and Meng, Cai and Bai, Xiangzhi},

journal={Computerized Medical Imaging and Graphics},

pages={102566},

year={2025},

publisher={Elsevier}

}