Guo-Stone / MambaMorph

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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/).

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}

}

About

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

License:MIT License


Languages

Language:Python 100.0%