diff --git a/modules/deeplearning/srcnn_cld_frac.py b/modules/deeplearning/srcnn_cld_frac.py index a62455e35be01b24a15210206f4bff9058ddfdcc..57b9f948db4dc97d78b19385f1bf3bb8ef14c9dc 100644 --- a/modules/deeplearning/srcnn_cld_frac.py +++ b/modules/deeplearning/srcnn_cld_frac.py @@ -790,9 +790,9 @@ 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, preds = nn.run_restore(directory, ckpt_dir) if out_file is not None: - np.save(out_file, [np.squeeze(labels_denorm), preds_denorm.argmax(axis=3)]) + np.save(out_file, [np.squeeze(labels), preds.argmax(axis=3)]) def run_evaluate_static(in_file, out_file, ckpt_dir):