modified vgg encoder

Yiming992 · Oct 24, 2018 · b811944 · b811944
1 parent fc09e4c
commit b811944
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/encoder_decoder_model/vgg_encoder.py b/encoder_decoder_model/vgg_encoder.py
@@ -58,11 +58,7 @@ def _conv_stage(self, input_tensor, k_size, out_dims, name,
 
             relu = self.relu(inputdata=bn, name='relu')
 
-            spatial_dropout = self.spatial_dropout(input_tensor=relu, keep_prob=0.5,
-                                                   is_training=self._is_training,
-                                                   name='spatial_dropout')
-
-            return spatial_dropout
+            return relu
 
     def _fc_stage(self, input_tensor, out_dims, name, use_bias=False):
         """
@@ -185,6 +181,7 @@ def encode(self, input_tensor, name):
 
         return ret
 
+
 if __name__ == '__main__':
     a = tf.placeholder(dtype=tf.float32, shape=[1, 2048, 2048, 3], name='input')
     encoder = VGG16Encoder(phase=tf.constant('train', dtype=tf.string))