diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 2f02c9a268b94ca3e3e91891b93466131b959e51..06abdbfd61021a079a80ff3ef85fea71cda0e092 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -419,14 +419,14 @@ class ESPCN: if NOISE_TRAINING: conv = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(conv) - conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding=padding, activation=activation)(conv) - conv = tf.keras.layers.BatchNormalization()(conv) - print(conv.shape) - conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=activation)(conv) conv = tf.keras.layers.BatchNormalization()(conv) print(conv.shape) + # conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=activation)(conv) + # conv = tf.keras.layers.BatchNormalization()(conv) + # print(conv.shape) + conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=None)(conv) conv = tf.keras.layers.BatchNormalization()(conv) print(conv.shape)