WIP: debugging

Author: dzenanz

monai/networks/layers/simplelayers.py

@@ -499,9 +499,11 @@ def median_filter(
     # map the local window to single vector
     conv = [F.conv1d, F.conv2d, F.conv3d][spatial_dims - 1]  # type: ignore
     # even-sized kernels are not supported
-    padding = [(k - 1) // 2 for k in kernel.shape[2:]]
-    padded_input: torch.Tensor = F.pad(in_tensor.reshape(oprod, 1, *sshape), pad=padding, mode="replicate")
-    features: torch.Tensor = conv(padded_input, kernel, stride=1, **kwargs)  # type: ignore
+    padding = [(k - 1) // 2 for k in kernel.shape[2:] for _ in range(2)]
+    padded_input: torch.Tensor = F.pad(in_tensor, pad=padding, mode="replicate")
+    psshape = padded_input.shape[-spatial_dims:]
+    reshaped_input = padded_input.reshape(oprod, 1, *psshape)
+    features: torch.Tensor = conv(reshaped_input, kernel, padding=0, stride=1, **kwargs)  # type: ignore
     features = features.view(oprod, -1, *sshape)  # type: ignore
 
     # compute the median along the feature axis