From 873fa5d4ee735e1ddccda9c9c3d211f710cff455 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Sat, 16 Jul 2022 10:18:34 +0100 Subject: [PATCH] update mednist Signed-off-by: Wenqi Li --- 2d_classification/mednist_tutorial.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/2d_classification/mednist_tutorial.ipynb b/2d_classification/mednist_tutorial.ipynb index f88556b2b0..ebdc92097a 100644 --- a/2d_classification/mednist_tutorial.ipynb +++ b/2d_classification/mednist_tutorial.ipynb @@ -360,7 +360,7 @@ "source": [ "train_transforms = Compose(\n", " [\n", - " LoadImage(),\n", + " LoadImage(image_only=True),\n", " AddChannel(),\n", " ScaleIntensity(),\n", " RandRotate(range_x=np.pi / 12, prob=0.5, keep_size=True),\n", @@ -370,7 +370,7 @@ ")\n", "\n", "val_transforms = Compose(\n", - " [LoadImage(), AddChannel(), ScaleIntensity()])\n", + " [LoadImage(image_only=True), AddChannel(), ScaleIntensity()])\n", "\n", "y_pred_trans = Compose([Activations(softmax=True)])\n", "y_trans = Compose([AsDiscrete(to_onehot=num_class)])" @@ -1340,7 +1340,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.13" + "version": "3.8.12" } }, "nbformat": 4,