Skip to content

Commit cffeaea

Browse files
Merge branch 'swin_unetr_btcv' of github.com:Project-MONAI/model-zoo into swin_unetr_btcv
2 parents 0a2d104 + cba0bad commit cffeaea

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

models/swin_unetr_btcv_segmentation/configs/metadata.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
"label_classes": "multi-channel data,0:background,1:spleen, 2:Right Kidney, 3:Left Kideny, 4:Gallbladder, 5:Esophagus, 6:Liver, 7:Stomach, 8:Aorta, 9:IVC, 10:Portal and Splenic Veins, 11:Pancreas, 12:Right adrenal gland, 13:Left adrenal gland",
2222
"pred_classes": "14 channels OneHot data, 0:background,1:spleen, 2:Right Kidney, 3:Left Kideny, 4:Gallbladder, 5:Esophagus, 6:Liver, 7:Stomach, 8:Aorta, 9:IVC, 10:Portal and Splenic Veins, 11:Pancreas, 12:Right adrenal gland, 13:Left adrenal gland",
2323
"eval_metrics": {
24-
"mean_dice": 0.8312
24+
"mean_dice": 0.8283
2525
},
2626
"intended_use": "This is an example, not to be used for diagnostic purposes",
2727
"references": [

models/swin_unetr_btcv_segmentation/docs/README.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,14 @@ The training data is from the [BTCV dataset](https://www.synapse.org/#!Synapse:s
1111
- Modality: CT
1212
- Size: 30 3D volumes (24 Training + 6 Testing)
1313

14+
The dataset format needs to be redefined using the following commands:
15+
16+
```
17+
mv RawData/Training/img/ RawData/imagesTr
18+
mv RawData/Training/label/ RawData/labelsTr
19+
mv RawData/Testing/img/ RawData/imagesTs
20+
```
21+
1422
## Training configuration
1523
The training was performed with at least 32GB-memory GPUs.
1624

@@ -28,7 +36,7 @@ A graph showing the validation mean Dice for 5000 epochs.
2836

2937
This model achieves the following Dice score on the validation data (our own split from the training dataset):
3038

31-
Mean Dice = 0.8312
39+
Mean Dice = 0.8283
3240

3341
Note that mean dice is computed in the original spacing of the input data.
3442
## commands example

0 commit comments

Comments
 (0)