diff --git a/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py b/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py index 65bb0cf387627aee576073de55332bfa47268980..5752ee8d8d77a2aa423afc7cd1262c8c5089f396 100644 --- a/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py +++ b/modules/deeplearning/cloud_fraction_fcn_abi_hkm_refl.py @@ -530,8 +530,10 @@ class SRCNN: 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_2') + num_filters_hkm = 32 + conv_hkm = build_conv2d_block(conv_hkm, num_filters_hkm, activation, 'Conv2D_1') + num_filters_hkm *= 2 + conv_hkm = build_conv2d_block(conv_hkm, num_filters_hkm, activation, 'Conv2D_2') conv_b = conv_b + conv_hkm print(conv_b.shape)