From 2ba17a1540499d85b033eb3dc6881cc93ef5d7bd Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 11:32:01 +0200 Subject: [PATCH] Fixed cleaning of labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/evaluate.py | 1 - src/twomartens/masterthesis/main.py | 8 ++++---- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/src/twomartens/masterthesis/evaluate.py b/src/twomartens/masterthesis/evaluate.py index 414d589..b0b93f2 100644 --- a/src/twomartens/masterthesis/evaluate.py +++ b/src/twomartens/masterthesis/evaluate.py @@ -48,7 +48,6 @@ def get_number_gt_per_class(labels: Sequence[Sequence[Sequence[int]]], # iterate over images for i in label_range: boxes = np.asarray(labels[i]) - print(boxes) # iterate over boxes in image for j in range(boxes.shape[0]): diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index af129c3..6b38e6e 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -208,10 +208,10 @@ def _ssd_test(args: argparse.Namespace) -> None: # get labels per batch _labels = np.asarray(pickle.load(file)) # exclude padded label entries - print(_labels.shape) - cleaned_labels = _labels[_labels[:, :, 0] != -1] - print(cleaned_labels.shape) - labels.extend(cleaned_labels) + for i in range(_labels.shape[0]): + image_labels = _labels[i] + image_labels = image_labels[image_labels[:, 0] != -1] + labels.append(image_labels) # store labels for later use with open(label_file, "wb") as file: pickle.dump(labels, file)