From 8a239128d5a535d49415038dcddf04aba1ea36de Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 10:52:21 +0200 Subject: [PATCH] Attempt to fix exclusion of padded label entries Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 0f93e94..384ee6b 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -209,7 +209,7 @@ def _ssd_test(args: argparse.Namespace) -> None: _labels = pickle.load(file) # exclude padded label entries real_labels = np.nonzero(_labels[:, :, 0] != -1) - labels.extend(_labels[real_labels]) + labels.extend(_labels[_labels[:, :, 0] != -1]) # store labels for later use with open(label_file, "wb") as file: pickle.dump(labels, file)