Set PSPNet final interpolation 'bilinear' (#262)

qubvel · Nov 21, 2019 · 5510008 · 5510008
1 parent d71189c
commit 5510008
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/segmentation_models/models/pspnet.py b/segmentation_models/models/pspnet.py
@@ -140,7 +140,7 @@ def build_psp(
         name='final_conv',
     )(x)
 
-    x = layers.UpSampling2D(final_upsampling_factor, name='final_upsampling')(x)
+    x = layers.UpSampling2D(final_upsampling_factor, name='final_upsampling', interpolation='bilinear')(x)
     x = layers.Activation(activation, name=activation)(x)
 
     model = models.Model(input_, x)