From 7f0e4e469c6457fb25f27ff4bf13901eda3a2124 Mon Sep 17 00:00:00 2001 From: swilson314 Date: Thu, 8 Dec 2022 16:34:35 -0800 Subject: [PATCH] 3D bug in EfficientNet Squeeze and Excitation layer believe the image_size = [1, 1] in _make_same_padder() should actually be [1, 1, 1] for the EfficientNet to support 3D. This is just by inspection, but I suspect this oversight came from extending 2D code. --- monai/networks/nets/efficientnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monai/networks/nets/efficientnet.py b/monai/networks/nets/efficientnet.py index eef3d68090..5a059409a6 100644 --- a/monai/networks/nets/efficientnet.py +++ b/monai/networks/nets/efficientnet.py @@ -161,9 +161,9 @@ def __init__( self._se_adaptpool = adaptivepool_type(1) num_squeezed_channels = max(1, int(in_channels * self.se_ratio)) self._se_reduce = conv_type(in_channels=oup, out_channels=num_squeezed_channels, kernel_size=1) - self._se_reduce_padding = _make_same_padder(self._se_reduce, [1, 1]) + self._se_reduce_padding = _make_same_padder(self._se_reduce, [1, 1, 1]) self._se_expand = conv_type(in_channels=num_squeezed_channels, out_channels=oup, kernel_size=1) - self._se_expand_padding = _make_same_padder(self._se_expand, [1, 1]) + self._se_expand_padding = _make_same_padder(self._se_expand, [1, 1, 1]) # Pointwise convolution phase final_oup = out_channels