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

snapshot...

parent ea97c348
Branches
No related tags found
No related merge requests found
...@@ -687,7 +687,6 @@ class SRCNN: ...@@ -687,7 +687,6 @@ class SRCNN:
preds = np.concatenate(self.test_preds) preds = np.concatenate(self.test_preds)
inputs = np.concatenate(self.test_input) inputs = np.concatenate(self.test_input)
cat_cld_frac = np.concatenate(self.test_cat_cf) cat_cld_frac = np.concatenate(self.test_cat_cf)
print(labels.shape, preds.shape, cat_cld_frac.shape, inputs.shape)
# labels = denormalize(labels, label_param, mean_std_dct) # labels = denormalize(labels, label_param, mean_std_dct)
# preds = denormalize(preds, label_param, mean_std_dct) # preds = denormalize(preds, label_param, mean_std_dct)
...@@ -897,12 +896,13 @@ def run_restore_static(directory, ckpt_dir, out_file=None): ...@@ -897,12 +896,13 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
if out_file is not None: if out_file is not None:
y_hi, x_hi = (Y_LEN // 4) + 1, (X_LEN // 4) + 1 y_hi, x_hi = (Y_LEN // 4) + 1, (X_LEN // 4) + 1
np.save(out_file, np.save(out_file,
[labels, preds, [labels[:, :, :, 0], preds[:, :, :, 0],
inputs[:, 1:y_hi, 1:x_hi, 0], inputs[:, 1:y_hi, 1:x_hi, 0],
descale(inputs[:, 1:y_hi, 1:x_hi, 1], 'refl_0_65um_nom', mean_std_dct), descale(inputs[:, 1:y_hi, 1:x_hi, 1], 'refl_0_65um_nom', mean_std_dct),
descale(inputs[:, 1:y_hi, 1:x_hi, 2], 'refl_0_65um_nom', mean_std_dct), descale(inputs[:, 1:y_hi, 1:x_hi, 2], 'refl_0_65um_nom', mean_std_dct),
inputs[:, 1:y_hi, 1:x_hi, 3], inputs[:, 1:y_hi, 1:x_hi, 3],
descale(inputs[:, 1:y_hi, 1:x_hi, 4], label_param, mean_std_dct)]) descale(inputs[:, 1:y_hi, 1:x_hi, 4], label_param, mean_std_dct),
cat_cld_frac[:, :, :]])
def run_evaluate_static(in_file, out_file, ckpt_dir): def run_evaluate_static(in_file, out_file, ckpt_dir):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment