snapshot...

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',