Changed name of val to test

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-06-13 17:03:29 +02:00
parent f9da5e8236
commit 232e531a9d
2 changed files with 16 additions and 16 deletions

View File

@ -237,14 +237,14 @@ def _ssd_evaluate(args: argparse.Namespace) -> None:
pickle.dump(results, file) 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": if args.network == "ssd" or args.network == "bayesian_ssd":
_ssd_val(args) _ssd_test(args)
elif args.network == "auto_encoder": 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 pickle
import os import os
@ -297,7 +297,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
nr_digits) 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 import data
from twomartens.masterthesis.aae import run from twomartens.masterthesis.aae import run
import tensorflow as tf import tensorflow as tf

View File

@ -42,22 +42,22 @@ def main() -> None:
prepare_parser = sub_parsers.add_parser("prepare", help="Prepare SceneNet RGB-D ground truth") 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") train_parser = sub_parsers.add_parser("train", help="Train a network")
evaluate_parser = sub_parsers.add_parser("evaluate", help="Evaluate 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 sub parsers
_build_prepare(prepare_parser) _build_prepare(prepare_parser)
_build_train(train_parser) _build_train(train_parser)
_build_val(val_parser) _build_test(test_parser)
_build_evaluate(evaluate_parser) _build_evaluate(evaluate_parser)
args = parser.parse_args() args = parser.parse_args()
if args.action == "train": if args.action == "train":
cli.train(args) cli.train(args)
elif args.action == "test": elif args.action == "evaluate":
cli.evaluate(args) cli.evaluate(args)
elif args.action == "val": elif args.action == "test":
cli.val(args) cli.test(args)
elif args.action == "prepare": elif args.action == "prepare":
cli.prepare(args) 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") 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 = parser.add_subparsers(dest="network")
sub_parsers.required = True 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") auto_encoder_parser = sub_parsers.add_parser("auto_encoder", help="Auto-encoder network")
# build sub parsers # build sub parsers
_build_ssd_val(ssd_bayesian_parser) _build_ssd_test(ssd_bayesian_parser)
_build_ssd_val(ssd_parser) _build_ssd_test(ssd_parser)
_build_auto_encoder_val(auto_encoder_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("--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("--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") 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") 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("--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("--weights_path", type=str, help="path to the weights directory")
parser.add_argument("--summary_path", type=str, help="path to the summaries directory") parser.add_argument("--summary_path", type=str, help="path to the summaries directory")