snapshot...

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)