Skip to content
Snippets Groups Projects
Commit a230af5f authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 7a231a6f
Branches
No related tags found
No related merge requests found
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment