diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 2e9372adc7711e943ef6fb7322bca814022e62be..7ee52b335c25cc444c011df44168b498fb5e78b9 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -375,13 +375,6 @@ class SRCNN: conv = conv + conv_b print(conv.shape) - # conv = tf.keras.layers.Conv2D(num_filters * (factor ** 2), 3, padding='same')(conv) - print(conv.shape) - - #conv = tf.nn.depth_to_space(conv, factor) - #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) print(self.logits.shape)