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

Improve filenames in measure_mapping

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-15 16:03:39 +02:00
parent 5d955d9c3c
commit 758c8a7574
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -135,6 +135,7 @@ def visualise(args: argparse.Namespace) -> None:
def measure_mapping(args: argparse.Namespace) -> None: def measure_mapping(args: argparse.Namespace) -> None:
import pickle import pickle
import os
from twomartens.masterthesis.ssd_keras.eval_utils import coco_utils from twomartens.masterthesis.ssd_keras.eval_utils import coco_utils
@ -142,8 +143,10 @@ def measure_mapping(args: argparse.Namespace) -> None:
instances = pickle.load(file) instances = pickle.load(file)
output_path = f"{args.output_path}/measure/{args.tarball_id}" output_path = f"{args.output_path}/measure/{args.tarball_id}"
os.makedirs(output_path, exist_ok=True)
annotation_file_train = f"{args.coco_path}/annotations/instances_train2014.json" annotation_file_train = f"{args.coco_path}/annotations/instances_train2014.json"
cats_to_classes, _, _, _ = coco_utils.get_coco_category_maps(annotation_file_train) cats_to_classes, _, _, _ = coco_utils.get_coco_category_maps(annotation_file_train)
nr_digits = _get_nr_digits(len(instances), 1)
for i, trajectory in enumerate(instances): for i, trajectory in enumerate(instances):
counts = {cat_id: 0 for cat_id in cats_to_classes.keys()} counts = {cat_id: 0 for cat_id in cats_to_classes.keys()}
@ -151,7 +154,7 @@ def measure_mapping(args: argparse.Namespace) -> None:
for instance in labels: for instance in labels:
counts[instance['coco_id']] += 1 counts[instance['coco_id']] += 1
with open(f"{output_path}/{i}.bin", "wb") as file: with open(f"{output_path}/{str(i).zfill(nr_digits)}.bin", "wb") as file:
pickle.dump(counts, file) pickle.dump(counts, file)