Skip to content

Commit 901964c

Browse files
authored
Update 3d-segmentation tutorials (#916)
* easy-integrate-bundle-v1 Signed-off-by: KumoLiu <yunl@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * add multigpu implementation Signed-off-by: KumoLiu <yunl@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * minor fix Signed-off-by: KumoLiu <yunl@nvidia.com> * del remove file Signed-off-by: KumoLiu <yunl@nvidia.com> Signed-off-by: KumoLiu <yunl@nvidia.com>
1 parent 5f97a73 commit 901964c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

3d_segmentation/challenge_baseline/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ During training, the top three models will be selected based on the per-epoch va
6464

6565
The training uses convenient file loading modules and a few intensity and spatial random augmentations using [MONAI](https://github.com/Project-MONAI/MONAI):
6666

67-
- `LoadImaged`, `AddChanneld`, `Orientationd`, `Spacingd`, `ScaleIntensityRanged`
67+
- `LoadImaged`, `Orientationd`, `Spacingd`, `ScaleIntensityRanged`
6868

6969
Load the image data into the LPS orientation (Left to right, Posterior to anterior, Superior to inferior), with a resolution of 1.25mm x 1.25mm x 5.00mm, and intensity between [-1000.0, 500.0] scaled to [0.0, 1.0].
7070

3d_segmentation/swin_unetr_brats21_segmentation_3d.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -479,7 +479,7 @@
479479
"torch.backends.cudnn.benchmark = True\n",
480480
"dice_loss = DiceLoss(to_onehot_y=False, sigmoid=True)\n",
481481
"post_sigmoid = Activations(sigmoid=True)\n",
482-
"post_pred = AsDiscrete(argmax=False, logit_thresh=0.5)\n",
482+
"post_pred = AsDiscrete(argmax=False, threshold=0.5)\n",
483483
"dice_acc = DiceMetric(\n",
484484
" include_background=True, reduction=MetricReduction.MEAN_BATCH, get_not_nans=True\n",
485485
")\n",

0 commit comments

Comments
 (0)