diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index d27784dec410d221dfcbf0a261023ca025af45b4..2fcf79060a2c6b3d2a5f490455483281598dd69e 100644 --- a/modules/deeplearning/espcn.py +++ b/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