diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 2153d54dd845073f271f41ec993fe0e68aeea468..2e9372adc7711e943ef6fb7322bca814022e62be 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -368,6 +368,8 @@ class SRCNN: 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, kernel_initializer='he_uniform', padding=padding)(conv_b) conv = conv + conv_b