diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 56cf121e059aee0509607d968256dca423979697..8eb388f253340ea51918e424f7a38e0d58f64b2d 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -356,7 +356,7 @@ class SRCNN: print('input: ', conv.shape) # conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, padding=padding)(input_2d) - conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, kernel_initializer='he_uniform', activation=activation, padding='VALID')(input_2d) + conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, kernel_initializer='he_uniform', activation=activation, padding='SAME')(input_2d) print(conv.shape) if NOISE_TRAINING: