Skip to content

Commit 902269d

Browse files
author
Oliver Rippel
committed
Remove BN + PRelu in segmentation layers
1 parent f913e27 commit 902269d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

unet3d/model/isensee2017.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from functools import partial
22

3-
from keras.layers import Input, LeakyReLU, Add, UpSampling3D, Activation, SpatialDropout3D
3+
from keras.layers import Input, LeakyReLU, Add, UpSampling3D, Activation, SpatialDropout3D, Conv3D
44
from keras.engine import Model
55
from keras.optimizers import Adam
66

@@ -61,7 +61,7 @@ def isensee2017_model(input_shape=(4, 128, 128, 128), n_base_filters=16, depth=5
6161
localization_output = create_localization_module(concatenation_layer, level_filters[level_number])
6262
current_layer = localization_output
6363
if level_number < n_segmentation_levels:
64-
segmentation_layers.insert(0, create_convolution_block(current_layer, n_filters=n_labels, kernel=(1, 1, 1)))
64+
segmentation_layers.insert(0, Conv3D(n_labels, (1, 1, 1))(current_layer))
6565

6666
output_layer = None
6767
for level_number in reversed(range(n_segmentation_levels)):

0 commit comments

Comments
 (0)