farrell236 commited on
Commit
cbc765c
·
verified ·
1 Parent(s): 9c89ced

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +47 -3
README.md CHANGED
@@ -1,3 +1,47 @@
1
- ---
2
- license: openmdw-1.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: openmdw-1.0
3
+ ---
4
+
5
+ # nnUNet_MSWAL
6
+
7
+ nnU-Net models for **MSWAL** lesion segmentation.
8
+
9
+ This repository contains nnU-Net models trained on the MSWAL dataset for 1000 and 4000 epochs.
10
+
11
+ Available model directories:
12
+ - `nnUNetTrainer__nnUNetResEncUNetLPlans__3d_fullres`
13
+ - `nnUNetTrainer_4000epochs__nnUNetResEncUNetLPlans__3d_fullres`
14
+
15
+ ## Inference
16
+
17
+ Place model in `nnUNet_results` directory; ensure nnU-Net environment variables are set; prediction can be run as follows:
18
+
19
+ ```bash
20
+ nnUNetv2_predict \
21
+ -i INPUT_FOLDER \
22
+ -o OUTPUT_FOLDER \
23
+ -d 201 \
24
+ -c 3d_fullres \
25
+ -f 0 1 2 3 4 \
26
+ # use nnUNetTrainer_4000epochs for the 4000-epoch model
27
+ -tr nnUNetTrainer \
28
+ -p nnUNetResEncUNetLPlans
29
+ ```
30
+
31
+ ## Reference
32
+
33
+ Please cite the original MSWAL work and refer to the official project resources.
34
+
35
+ ```bibtex
36
+ @inproceedings{wu2025mswal,
37
+ title={Mswal: 3d multi-class segmentation of whole abdominal lesions dataset},
38
+ 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},
39
+ booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
40
+ pages={378--388},
41
+ year={2025},
42
+ organization={Springer}
43
+ }
44
+ ```
45
+
46
+ Official MSWAL repository:
47
+ https://github.com/haochen-MBZUAI/MSWAL-