diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 0574496..5ed0376 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -237,14 +237,14 @@ def _ssd_evaluate(args: argparse.Namespace) -> None: pickle.dump(results, file) -def val(args: argparse.Namespace) -> None: +def test(args: argparse.Namespace) -> None: if args.network == "ssd" or args.network == "bayesian_ssd": - _ssd_val(args) + _ssd_test(args) elif args.network == "auto_encoder": - _auto_encoder_val(args) + _auto_encoder_test(args) -def _ssd_val(args: argparse.Namespace) -> None: +def _ssd_test(args: argparse.Namespace) -> None: import pickle import os @@ -297,7 +297,7 @@ def _ssd_val(args: argparse.Namespace) -> None: nr_digits) -def _auto_encoder_val(args: argparse.Namespace) -> None: +def _auto_encoder_test(args: argparse.Namespace) -> None: from twomartens.masterthesis import data from twomartens.masterthesis.aae import run import tensorflow as tf diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 3faa61d..c2d606d 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -42,22 +42,22 @@ def main() -> None: prepare_parser = sub_parsers.add_parser("prepare", help="Prepare SceneNet RGB-D ground truth") train_parser = sub_parsers.add_parser("train", help="Train a network") evaluate_parser = sub_parsers.add_parser("evaluate", help="Evaluate a network") - val_parser = sub_parsers.add_parser("val", help="Validate a network") + test_parser = sub_parsers.add_parser("test", help="Test a network") # build sub parsers _build_prepare(prepare_parser) _build_train(train_parser) - _build_val(val_parser) + _build_test(test_parser) _build_evaluate(evaluate_parser) args = parser.parse_args() if args.action == "train": cli.train(args) - elif args.action == "test": + elif args.action == "evaluate": cli.evaluate(args) - elif args.action == "val": - cli.val(args) + elif args.action == "test": + cli.test(args) elif args.action == "prepare": cli.prepare(args) @@ -103,7 +103,7 @@ def _build_auto_encoder_train(parser: argparse.ArgumentParser) -> None: parser.add_argument("iteration", type=int, help="the training iteration") -def _build_val(parser: argparse.ArgumentParser) -> None: +def _build_test(parser: argparse.ArgumentParser) -> None: sub_parsers = parser.add_subparsers(dest="network") sub_parsers.required = True @@ -112,12 +112,12 @@ def _build_val(parser: argparse.ArgumentParser) -> None: auto_encoder_parser = sub_parsers.add_parser("auto_encoder", help="Auto-encoder network") # build sub parsers - _build_ssd_val(ssd_bayesian_parser) - _build_ssd_val(ssd_parser) - _build_auto_encoder_val(auto_encoder_parser) + _build_ssd_test(ssd_bayesian_parser) + _build_ssd_test(ssd_parser) + _build_auto_encoder_test(auto_encoder_parser) -def _build_ssd_val(parser: argparse.ArgumentParser) -> None: +def _build_ssd_test(parser: argparse.ArgumentParser) -> None: parser.add_argument("--coco_path", type=str, help="the path to the COCO data set") parser.add_argument("--weights_path", type=str, help="path to the weights directory") parser.add_argument("--ground_truth_path", type=str, help="path to the prepared ground truth directory") @@ -127,7 +127,7 @@ def _build_ssd_val(parser: argparse.ArgumentParser) -> None: parser.add_argument("train_iteration", type=int, help="the train iteration") -def _build_auto_encoder_val(parser: argparse.ArgumentParser) -> None: +def _build_auto_encoder_test(parser: argparse.ArgumentParser) -> None: parser.add_argument("--coco_path", type=str, help="the path to the COCO data set") parser.add_argument("--weights_path", type=str, help="path to the weights directory") parser.add_argument("--summary_path", type=str, help="path to the summaries directory")