From c35af5fc331c6686b07120beaedde3666392d6f1 Mon Sep 17 00:00:00 2001 From: Rich <33289025+rijobro@users.noreply.github.com> Date: Tue, 12 Jan 2021 14:04:40 +0000 Subject: [PATCH 1/2] fix bug Signed-off-by: Richard Brown <33289025+rijobro@users.noreply.github.com> --- monai/visualize/occlusion_sensitivity.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monai/visualize/occlusion_sensitivity.py b/monai/visualize/occlusion_sensitivity.py index 6dc4de154f..559a7f371d 100644 --- a/monai/visualize/occlusion_sensitivity.py +++ b/monai/visualize/occlusion_sensitivity.py @@ -303,7 +303,8 @@ def __call__( # type: ignore # upsample if self.upsampler is not None: - if np.any(output_im_shape != x.shape[1:]): + assert(len(sensitivity_ims_list[i].shape) == len(x.shape)) + if np.any(sensitivity_ims_list[i].shape != x.shape): img_spatial = tuple(output_im_shape[1:]) sensitivity_ims_list[i] = self.upsampler(img_spatial)(sensitivity_ims_list[i]) From 54c049e2191baad5ba673b869a95e123dea683e4 Mon Sep 17 00:00:00 2001 From: Richard Brown <33289025+rijobro@users.noreply.github.com> Date: Tue, 12 Jan 2021 16:10:31 +0000 Subject: [PATCH 2/2] autofix Signed-off-by: Richard Brown <33289025+rijobro@users.noreply.github.com> --- monai/visualize/occlusion_sensitivity.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/visualize/occlusion_sensitivity.py b/monai/visualize/occlusion_sensitivity.py index 559a7f371d..bb9ef59e5c 100644 --- a/monai/visualize/occlusion_sensitivity.py +++ b/monai/visualize/occlusion_sensitivity.py @@ -303,7 +303,7 @@ def __call__( # type: ignore # upsample if self.upsampler is not None: - assert(len(sensitivity_ims_list[i].shape) == len(x.shape)) + assert len(sensitivity_ims_list[i].shape) == len(x.shape) if np.any(sensitivity_ims_list[i].shape != x.shape): img_spatial = tuple(output_im_shape[1:]) sensitivity_ims_list[i] = self.upsampler(img_spatial)(sensitivity_ims_list[i])