Improved formatting of main

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-06-04 10:36:18 +02:00
parent 9851cef758
commit 13cecd2254

View File

@ -360,7 +360,8 @@ def _prepare(args: argparse.Namespace) -> None:
from twomartens.masterthesis import data from twomartens.masterthesis import data
file_names_photos, file_names_instances, instances = data.prepare_scenenet_val(args.scenenet_path, args.protobuf_path) file_names_photos, file_names_instances, instances = data.prepare_scenenet_val(args.scenenet_path,
args.protobuf_path)
with open(f"{args.ground_truth_path}/photo_paths.bin", "wb") as file: with open(f"{args.ground_truth_path}/photo_paths.bin", "wb") as file:
pickle.dump(file_names_photos, file) pickle.dump(file_names_photos, file)
with open(f"{args.ground_truth_path}/instance_paths.bin", "wb") as file: with open(f"{args.ground_truth_path}/instance_paths.bin", "wb") as file: