diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py index 347029dbc3963220c7eb2a2c2fe65d01a56ebf86..30fad5c677b0fade9f8cf4ba7206e8e754f3c604 100644 --- a/modules/deeplearning/cloud_fraction_fcn_abi.py +++ b/modules/deeplearning/cloud_fraction_fcn_abi.py @@ -82,11 +82,16 @@ KERNEL_SIZE = 3 X_LEN = Y_LEN = 128 if KERNEL_SIZE == 3: - slc_x = slice(0, int(X_LEN/4) + 2) - slc_y = slice(0, int(Y_LEN/4) + 2) + slc_x = slice(0, X_LEN // 4 + 2) + slc_y = slice(0, Y_LEN // 4 + 2) x_64 = slice(4, X_LEN + 4) y_64 = slice(4, Y_LEN + 4) # ---------------------------------------- +if KERNEL_SIZE == 1: + slc_x = slice(0, X_LEN // 4) + slc_y = slice(0, Y_LEN // 4) + x_64 = slice(0, X_LEN) + y_64 = slice(0, Y_LEN) def build_residual_conv2d_block(conv, num_filters, block_name, activation=tf.nn.relu, padding='SAME',