snapshot...

modules/deeplearning/espcn.py

@@ -343,7 +343,7 @@ class ESPCN:
         activation = tf.nn.leaky_relu
         momentum = 0.99
 
-        num_filters = 32
+        num_filters = 128
 
         input_2d = self.inputs[0]
         print('input: ', input_2d.shape)
@@ -366,6 +366,8 @@ class ESPCN:
 
         conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4')
 
+        conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5')
+
         conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding)(conv_b)
 
         conv = conv + conv_b