snapshot...

modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py

@@ -543,11 +543,11 @@ class SRCNN:
 
         scale = 0.2
 
-        conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_1', kernel_size=KERNEL_SIZE, scale=scale)
+        conv_b = build_residual_conv2d_block(conv_b, num_filters + num_filters_hkm, 'Residual_Block_1', kernel_size=KERNEL_SIZE, scale=scale)
 
-        conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_2', kernel_size=KERNEL_SIZE, scale=scale)
+        conv_b = build_residual_conv2d_block(conv_b, num_filters + num_filters_hkm, 'Residual_Block_2', kernel_size=KERNEL_SIZE, scale=scale)
 
-        conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_3', kernel_size=KERNEL_SIZE, scale=scale)
+        conv_b = build_residual_conv2d_block(conv_b, num_filters + num_filters_hkm, 'Residual_Block_3', kernel_size=KERNEL_SIZE, scale=scale)
 
         #conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4', kernel_size=KERNEL_SIZE, scale=scale)
 
@@ -555,7 +555,7 @@ class SRCNN:
 
         #conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_6', kernel_size=KERNEL_SIZE, scale=scale)
 
-        conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
+        conv_b = tf.keras.layers.Conv2D(num_filters + num_filters_hkm, kernel_size=KERNEL_SIZE, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
 
         # conv = conv + conv_b
         conv = conv_b