diff --git a/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py b/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py index e5d841a46500943ac859260a77e5714ac29775d5..4954afb54f3b938db63a361542beb013687c53f9 100644 --- a/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py +++ b/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py @@ -527,6 +527,14 @@ class SRCNN: conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, kernel_initializer='he_uniform', activation=activation, padding='VALID')(input_2d) print(conv.shape) + conv_hkm = conv_hkm_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, kernel_initializer='he_uniform', activation=activation, padding='VALID')(input_hkm_2d) + print(conv_hkm.shape) + + conv_hkm = build_conv2d_block(conv_hkm, num_filters, activation, 'Conv2D_1') + conv_hkm = build_conv2d_block(conv_hkm, num_filters, activation, 'Conv2D_1') + + conv_b = conv_b + conv_hkm + # if NOISE_TRAINING: # conv = conv_b = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(conv)