diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 3a4dfd0c776344b726152816df7491d7541440d3..6d93edc7c11ced3ac2f2d873a469c52f178df849 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -447,15 +447,15 @@ class SRCNN: scale = 0.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_1', kernel_size=3, scale=scale) - 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_2', kernel_size=3, scale=scale) - #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_3', kernel_size=3, scale=scale) - #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_4', kernel_size=3, scale=scale) - #conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5', scale=scale) + #conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5', kernel_size=3, scale=scale) conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)