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

snapshot...

parent 8ee50caa
Branches
No related tags found
No related merge requests found
...@@ -253,8 +253,11 @@ class SRCNN: ...@@ -253,8 +253,11 @@ class SRCNN:
idx = params.index(param) idx = params.index(param)
tmp = input_data[:, idx, :, :] tmp = input_data[:, idx, :, :]
tmp = np.where(np.isnan(tmp), 0, tmp) tmp = np.where(np.isnan(tmp), 0, tmp)
tmp = tmp[:, self.slc_y_m, self.slc_x_m] # tmp = tmp[:, self.slc_y_m, self.slc_x_m]
tmp = self.upsample(tmp) # tmp = self.upsample(tmp)
tmp = upsample_mean(tmp)
tmp = tmp[:, self.slc_y, self.slc_x]
# ---------------
tmp = normalize(tmp, param, mean_std_dct) tmp = normalize(tmp, param, mean_std_dct)
data_norm.append(tmp) data_norm.append(tmp)
...@@ -280,8 +283,11 @@ class SRCNN: ...@@ -280,8 +283,11 @@ class SRCNN:
# --------------------------------------------------- # ---------------------------------------------------
tmp = input_label[:, label_idx_i, :, :] tmp = input_label[:, label_idx_i, :, :]
tmp = np.where(np.isnan(tmp), 0, tmp) tmp = np.where(np.isnan(tmp), 0, tmp)
tmp = tmp[:, self.slc_y_2, self.slc_x_2] # tmp = tmp[:, self.slc_y_2, self.slc_x_2]
tmp = self.upsample(tmp) # tmp = self.upsample(tmp)
tmp = upsample_mean(tmp)
tmp = tmp[:, self.slc_y, self.slc_x]
# ----------------
tmp = normalize(tmp, label_param, mean_std_dct) tmp = normalize(tmp, label_param, mean_std_dct)
data_norm.append(tmp) data_norm.append(tmp)
# --------- # ---------
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment