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

Updated prepare action to use new config

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-04 16:25:57 +02:00
parent f6c66d1d34
commit 3354a033fe
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/twomartens/masterthesis/cli.py
View File

@ -45,13 +45,19 @@ def prepare(args: argparse.Namespace) -> None:
from twomartens.masterthesis import data
file_names_photos, file_names_instances, instances = data.prepare_scenenet_data(args.scenenet_path,
args.protobuf_path)
with open(f"{args.ground_truth_path}/photo_paths.bin", "wb") as file:
file_names_photos, file_names_instances, \
instances = data.prepare_scenenet_data(conf.get_property("Paths.scenenet"),
args.protobuf_path)
with open(f"{conf.get_property('Paths.scenenet_gt')}/"
f"{args.ground_truth_path}/photo_paths.bin", "wb") as file:
pickle.dump(file_names_photos, file)
with open(f"{args.ground_truth_path}/instance_paths.bin", "wb") as file:
with open(f"{conf.get_property('Paths.scenenet_gt')}/"
f"{args.ground_truth_path}/instance_paths.bin", "wb") as file:
pickle.dump(file_names_instances, file)
with open(f"{args.ground_truth_path}/instances.bin", "wb") as file:
with open(f"{conf.get_property('Paths.scenenet_gt')}/"
f"{args.ground_truth_path}/instances.bin", "wb") as file:
pickle.dump(instances, file)

1
src/twomartens/masterthesis/config.py
View File

@ -36,6 +36,7 @@ _CONFIG_PROPS = {
"Paths": {
"coco": (str, ""),
"scenenet": (str, ""),
"scenenet_gt": (str, ""),
"scenenet_gt_train": (str, ""),
"scenenet_gt_val": (str, ""),
"scenenet_gt_test": (str, ""),