search

Guo-Stone / MambaMorph

MambaMorph: a Mamba-based Framework for Medical MR-CT Deformable Registration
MIT License
72 stars 10 forks source link

readme

MambaMorph: a Mamba-based Framework for Medical MR-CT Deformable Registration

Tutorial

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

Framework

MambaMorph-nocl-v2-github

Result

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/). The SR-Reg dataset is undergoing careful processing and will be published with the next version of MambaMorph.

Data sample

data-sample

Contact

guotao@buaa.edu.cn

Paper

https://arxiv.org/abs/2401.13934

Citation

@article{guo2024mambamorph,

title={MambaMorph: a Mamba-based Framework for Medical MR-CT Deformable Registration},

author={Guo, Tao and Wang, Yinuo and Shu, Shihao and Chen, Diansheng and Tang, Zhouping and Meng, Cai and Bai, Xiangzhi},

journal={arXiv preprint arXiv:2401.13934},

year={2024}

}