snapshot...

modules/deeplearning/cloud_fraction_fcn_abi.py

@@ -71,13 +71,13 @@ print('label_param: ', label_param)
 
 KERNEL_SIZE = 3  # target size: (128, 128)
 N_X = N_Y = 1
-X_LEN = Y_LEN = 128
+X_LEN = Y_LEN = 64
 
 if KERNEL_SIZE == 3:
-    slc_x = slice(1, int((N_X*X_LEN)/2) + 3)
-    slc_y = slice(1, int((N_Y*Y_LEN)/2) + 3)
-    x_128 = slice(4, N_X*X_LEN + 4)
-    y_128 = slice(4, N_Y*Y_LEN + 4)
+    slc_x = slice(0, int((N_X*X_LEN)/4) + 2)
+    slc_y = slice(0, int((N_Y*Y_LEN)/4) + 2)
+    x_64 = slice(4, N_X*X_LEN + 4)
+    y_64 = slice(4, N_Y*Y_LEN + 4)
 # elif KERNEL_SIZE == 5: These no longer apply here
 #     slc_x = slice(3, 135)
 #     slc_y = slice(3, 135)
@@ -348,7 +348,7 @@ class SRCNN:
         # -----------------------------------------------------
         # -----------------------------------------------------
         label = input_label[:, label_idx_i, :, :]
-        label = label[:, y_128, x_128]
+        label = label[:, y_64, x_64]
         if NumClasses == 5:
             label = get_label_data_5cat(label)
         else: