diff --git a/modules/deeplearning/espcn_l1b_l2.py b/modules/deeplearning/espcn_l1b_l2.py index a9b3c3916277764ab571e5889f6a8bfaad614cac..8a85149516edb452fe01f2d971f063a06280fa37 100644 --- a/modules/deeplearning/espcn_l1b_l2.py +++ b/modules/deeplearning/espcn_l1b_l2.py @@ -72,8 +72,10 @@ y_134_2 = y_134[2:133:2] slc_x = slice(3, 131) slc_y = slice(3, 131) -slc_x_2 = slice(3, 131, 2) -slc_y_2 = slice(3, 131, 2) +#slc_x_2 = slice(3, 131, 2) +#slc_y_2 = slice(3, 131, 2) +slc_x_2 = slice(2, 133, 2) +slc_y_2 = slice(2, 133, 2) def build_residual_conv2d_block(conv, num_filters, block_name, activation=tf.nn.leaky_relu, padding='SAME', scale=None): @@ -388,8 +390,8 @@ class ESPCN: # conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, kernel_initializer=kernel_initializer)(conv_b) - # conv = conv + conv_b - conv = conv_b + conv = conv + conv_b + # conv = conv_b print(conv.shape) conv = tf.keras.layers.Conv2D(IMG_DEPTH * (factor ** 2), 3, padding='same', activation=activation)(conv)