From b41b58930d150707a1d1aff24cf3b96fc89b82d5 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 13 Jun 2019 17:20:50 +0200 Subject: [PATCH] Adapted evaluation code to new file structure of saved labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 5ed0376..50dbd4f 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -174,12 +174,9 @@ def _ssd_evaluate(args: argparse.Namespace) -> None: for filename in files: with open(filename, "rb") as file: # get labels per batch - _labels = np.asarray(pickle.load(file)) - # exclude padded label entries - for i in range(_labels.shape[0]): - image_labels = _labels[i] - image_labels = image_labels[image_labels[:, 0] != -1] - labels.append(image_labels) + label_dict = pickle.load(file) + labels.extend(label_dict['labels']) + # store labels for later use with open(label_file, "wb") as file: pickle.dump(labels, file)