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

snapshot...

parent a4321ec6
Branches
No related tags found
No related merge requests found
......@@ -620,10 +620,11 @@ class SRCNN:
preds = np.concatenate(self.test_preds)
print(labels.shape, preds.shape)
labels_denorm = denormalize(labels, label_param, mean_std_dct)
preds_denorm = denormalize(preds, label_param, mean_std_dct)
# labels_denorm = denormalize(labels, label_param, mean_std_dct)
# preds_denorm = denormalize(preds, label_param, mean_std_dct)
return labels_denorm, preds_denorm
# return labels_denorm, preds_denorm
return labels, preds
def do_evaluate(self, inputs, ckpt_dir):
......@@ -674,9 +675,11 @@ class SRCNN:
def run_restore_static(directory, ckpt_dir, out_file=None):
nn = SRCNN()
labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir)
# labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir)
labels, preds = nn.run_restore(directory, ckpt_dir)
if out_file is not None:
np.save(out_file, [labels_denorm, preds_denorm])
# np.save(out_file, [labels_denorm, preds_denorm])
np.save(out_file, [labels, preds])
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