2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added config to CLI parsers

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-04 14:25:58 +02:00
parent bd499425ed
commit 136b52a7f4
1 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/twomartens/masterthesis/main.py
View File

@ -39,6 +39,7 @@ def main() -> None:
sub_parsers = parser.add_subparsers(dest="action") sub_parsers = parser.add_subparsers(dest="action")
sub_parsers.required = True sub_parsers.required = True
config_parser = sub_parsers.add_parser("config", help="Get and set config values")
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")
@ -47,6 +48,7 @@ def main() -> None:
measure_parser = sub_parsers.add_parser("measure_mapping", help="Measure the number of instances per COCO category") measure_parser = sub_parsers.add_parser("measure_mapping", help="Measure the number of instances per COCO category")
# build sub parsers # build sub parsers
_build_config(config_parser)
_build_prepare(prepare_parser) _build_prepare(prepare_parser)
_build_train(train_parser) _build_train(train_parser)
_build_test(test_parser) _build_test(test_parser)
@ -56,7 +58,9 @@ def main() -> None:
args = parser.parse_args() args = parser.parse_args()
if args.action == "train": if args.action == "config":
cli.config(args)
elif args.action == "train":
cli.train(args) cli.train(args)
elif args.action == "evaluate": elif args.action == "evaluate":
cli.evaluate(args) cli.evaluate(args)
@ -70,6 +74,27 @@ def main() -> None:
cli.measure_mapping(args) cli.measure_mapping(args)
def _build_config(parser: argparse.ArgumentParser) -> None:
sub_parsers = parser.add_subparsers(dest="action")
sub_parsers.required = True
get_parser = sub_parsers.add_parser("get", help="Get a config value")
set_parser = sub_parsers.add_parser("set", help="Set a config value")
sub_parsers.add_parser("list", help="List all config values")
_build_config_get(get_parser)
_build_config_set(set_parser)
def _build_config_get(parser: argparse.ArgumentParser) -> None:
parser.add_argument("property", type=str, help="config property to retrieve")
def _build_config_set(parser: argparse.ArgumentParser) -> None:
parser.add_argument("property", type=str, help="config property to set")
parser.add_argument("value", type=str, help="new value for config property")
def _build_prepare(parser: argparse.ArgumentParser) -> None: def _build_prepare(parser: argparse.ArgumentParser) -> None:
parser.add_argument("scenenet_path", type=str, help="the path to the SceneNet RGB-D data set") parser.add_argument("scenenet_path", type=str, help="the path to the SceneNet RGB-D data set")
parser.add_argument("protobuf_path", type=str, help="the path to the SceneNet RGB-D protobuf file") parser.add_argument("protobuf_path", type=str, help="the path to the SceneNet RGB-D protobuf file")