snapshot...

modules/deeplearning/cloud_opd_srcnn_abi_v2.py

@@ -740,6 +740,7 @@ class SRCNN:
         print(refl.shape)
         refl = refl[1912:3512, 1912:3512]
         LEN_Y, LEN_X = refl.shape
+        LEN_Y_in, LEN_X_in = LEN_Y, LEN_X
         print(refl.shape)
         bt = get_grid_values_all(h5f, 'temp_11_0um_nom')
         bt = bt[1912:3512, 1912:3512]
@@ -766,7 +767,7 @@ class SRCNN:
         print('inference time: ', (t1 - t0))
         print(cld_opd_sres.shape)
 
-        cld_opd_sres_out = np.zeros((LEN_Y, LEN_X), dtype=np.int8)
+        cld_opd_sres_out = np.zeros((LEN_Y_in, LEN_X_in), dtype=np.int8)
         border = int((KERNEL_SIZE - 1) / 2)
         cld_opd_sres_out[border:LEN_Y - border, border:LEN_X - border] = cld_opd_sres[0, :, :, 0]