diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 8eb388f253340ea51918e424f7a38e0d58f64b2d..dc9c62a887cd129b1e05d07acc4e2f585c685e60 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -362,15 +362,17 @@ class SRCNN: if NOISE_TRAINING: conv = conv_b = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(conv) - conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_1') + scale = 0.1 - conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_2') + conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_1', scale=scale) - conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_3') + conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_2', scale=scale) - 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_3', scale=scale) - conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5') + conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4', scale=scale) + + conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5', scale=scale) conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, kernel_initializer='he_uniform', padding=padding)(conv_b)