From b649c72b91e5b1899b03039998a4686a4c9f6777 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Fri, 26 Mar 2021 07:24:19 +0000 Subject: [PATCH 1/2] fixes #161 Signed-off-by: Wenqi Li --- modules/transforms_demo_2d.ipynb | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/modules/transforms_demo_2d.ipynb b/modules/transforms_demo_2d.ipynb index 138e1b929d..d57c8ae2c3 100644 --- a/modules/transforms_demo_2d.ipynb +++ b/modules/transforms_demo_2d.ipynb @@ -288,7 +288,11 @@ "# convert both image and segmentation using different interpolation mode\n", "new_img = affine(im_data, (300, 400), mode=\"bilinear\")\n", "new_seg = affine(seg_data, (300, 400), mode=\"nearest\")\n", - "print(new_img.shape, new_seg.shape)" + "print(new_img[0].shape, new_seg[0].shape)\n", + "print(\"affine:\", new_img[1])\n", + "print(\"affine:\", new_seg[1])\n", + "new_img = new_img[0]\n", + "new_seg = new_seg[0]" ] }, { @@ -427,7 +431,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.5" + "version": "3.7.10" } }, "nbformat": 4, From dfbedd0d43cafcdf067ce283d5ffa58d35095d05 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Fri, 26 Mar 2021 09:02:08 +0000 Subject: [PATCH 2/2] based on comments Signed-off-by: Wenqi Li --- modules/transforms_demo_2d.ipynb | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/modules/transforms_demo_2d.ipynb b/modules/transforms_demo_2d.ipynb index d57c8ae2c3..af17b76b32 100644 --- a/modules/transforms_demo_2d.ipynb +++ b/modules/transforms_demo_2d.ipynb @@ -286,13 +286,9 @@ " device=torch.device(\"cuda:0\"),\n", ")\n", "# convert both image and segmentation using different interpolation mode\n", - "new_img = affine(im_data, (300, 400), mode=\"bilinear\")\n", - "new_seg = affine(seg_data, (300, 400), mode=\"nearest\")\n", - "print(new_img[0].shape, new_seg[0].shape)\n", - "print(\"affine:\", new_img[1])\n", - "print(\"affine:\", new_seg[1])\n", - "new_img = new_img[0]\n", - "new_seg = new_seg[0]" + "new_img, _ = affine(im_data, (300, 400), mode=\"bilinear\")\n", + "new_seg, _ = affine(seg_data, (300, 400), mode=\"nearest\")\n", + "print(new_img.shape, new_seg.shape)" ] }, {