Please put the split file in ./dataset/dataset_0.json in the root folder of dataset
--data
----BTCV
------imagesTr
------imagesTs
------labelsTr
------dataset_0.json
Please find the BTCV dataset setup instructions from UNETR repor at this Link
Checkpoints available at
| Syntax | Feature Size | Dice | Weight |
|---|---|---|---|
| Mamba3D-S/16 | 32 | 83.1 | weight |
| Mamba3D-S+/16 | 32 | 83.9 | weight |
| Mamba3D-B/16 | 32 | 82.7 | weight |
| Mamba3D-B/16 | 64 | 84.7 | weight |
See exps folder for training scripts
python main.py --val --data_dir=<RawDataPath> --use_mamba --hidden_size <hidden size> --load_ckpt <ckpt> --resume_ckpt --feature_size <feature size>
where the hidden size for Mamba-ND-S, Mamba-ND-S+, and Mamba-ND-B is 384,512,768