File size: 1,357 Bytes
cbc765c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | ---
license: openmdw-1.0
---
# nnUNet_MSWAL
nnU-Net models for **MSWAL** lesion segmentation.
This repository contains nnU-Net models trained on the MSWAL dataset for 1000 and 4000 epochs.
Available model directories:
- `nnUNetTrainer__nnUNetResEncUNetLPlans__3d_fullres`
- `nnUNetTrainer_4000epochs__nnUNetResEncUNetLPlans__3d_fullres`
## Inference
Place model in `nnUNet_results` directory; ensure nnU-Net environment variables are set; prediction can be run as follows:
```bash
nnUNetv2_predict \
-i INPUT_FOLDER \
-o OUTPUT_FOLDER \
-d 201 \
-c 3d_fullres \
-f 0 1 2 3 4 \
# use nnUNetTrainer_4000epochs for the 4000-epoch model
-tr nnUNetTrainer \
-p nnUNetResEncUNetLPlans
```
## Reference
Please cite the original MSWAL work and refer to the official project resources.
```bibtex
@inproceedings{wu2025mswal,
title={Mswal: 3d multi-class segmentation of whole abdominal lesions dataset},
author={Wu, Zhaodong and Zhao, Qiaochu and Hu, Ming and Li, Yulong and Xue, Haochen and Jiang, Zhengyong and Stefanidis, Angelos and Wang, Qiufeng and Razzak, Imran and Ge, Zongyuan and others},
booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
pages={378--388},
year={2025},
organization={Springer}
}
```
Official MSWAL repository:
https://github.com/haochen-MBZUAI/MSWAL-
|