c9fb02940b16716901d0a2eef95abab827c25896,src/koalarization/evaluate.py,,,#,9

Before Change

        saver.restore(sess, latest_checkpoint)
        print(" done!")
    else:
        print("No checkpoint found in:", checkpoint_paths)

    // Evaluation (epoch=-1 to say that this is an evaluation after training)
    res = sess.run(evaluations_ops)
    print("Cost: {}".format(res["cost"]))

After Change

from .training_utils import evaluation_pipeline, \
    checkpointing_system, plot_evaluation, metrics_system

parser = argparse.ArgumentParser(description="Eval")
parser.add_argument("tfrecords",
                    type=str,
                    metavar="TFRECORDS_DIR",
                    help="evaluate using all tfrecords in TFRECORDS_DIR")
parser.add_argument("output",
                    type=str,
                    metavar="OUR_DIR",
                    help="use OUR_DIR to load checkpoints and write images")
parser.add_argument("--run-id",
                    required=True,
                    type=str,
                    metavar="RUN_ID",
                    help="load checkpoint from the run RUN_ID")
args = parser.parse_args()
dir_tfrecords = Path(args.tfrecords).expanduser().resolve().as_posix()
dir_output = Path(args.output).expanduser().resolve().joinpath(args.run_id).as_posix()

// PARAMETERS

Instances