diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index f0b66dfc112fa7dff1fd743f07ea5072ba976e43..1ab09707bab8af737f859821abb0326a4ab54feb 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -375,7 +375,7 @@ class ESPCN: print(conv.shape) #conv = tf.nn.depth_to_space(conv, factor) - conv = tf.keras.layers.Conv2DTranspose(num_filters * (factor ** 2), 3, padding='same')(conv) + #conv = tf.keras.layers.Conv2DTranspose(num_filters * (factor ** 2), 3, padding='same')(conv) print(conv.shape) self.logits = tf.keras.layers.Conv2D(1, kernel_size=3, strides=1, padding=padding, name='regression')(conv)