From 8a96d2867c516880bc7e5ce83d3a13618e79fccb Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 14 Dec 2022 11:35:34 -0600 Subject: [PATCH] snapshot.. --- modules/deeplearning/srcnn_l1b_l2.py | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index fe437eea..287aafa9 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -63,20 +63,21 @@ print('data_params: ', data_params) print('label_param: ', label_param) KERNEL_SIZE = 3 # target size: (128, 128) +N = 1 if KERNEL_SIZE == 3: - slc_x = slice(2, 132) - slc_y = slice(2, 132) - slc_x_2 = slice(1, 134, 2) - slc_y_2 = slice(1, 134, 2) - x_2 = np.arange(67) - y_2 = np.arange(67) - t = np.arange(0, 67, 0.5) - s = np.arange(0, 67, 0.5) - x_k = slice(1, 131) - y_k = slice(1, 131) - x_128 = slice(3, 131) - y_128 = slice(3, 131) + slc_x = slice(2, N*128 + 4) + slc_y = slice(2, N*128 + 4) + slc_x_2 = slice(1, N*128 + 6, 2) + slc_y_2 = slice(1, N*128 + 6, 2) + x_2 = np.arange(int((N*128)/2) + 3) + y_2 = np.arange(int((N*128)/2) + 3) + t = np.arange(0, int((N*128)/2) + 3, 0.5) + s = np.arange(0, int((N*128)/2) + 3, 0.5) + x_k = slice(1, N*128 + 3) + y_k = slice(1, N*128 + 3) + x_128 = slice(3, N*128 + 3) + y_128 = slice(3, N*128 + 3) elif KERNEL_SIZE == 5: slc_x = slice(3, 135) slc_y = slice(3, 135) -- GitLab