From 970ed8d6c0060c2c55733383574b9aaa4bc1a34e Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 11:14:50 +0200 Subject: [PATCH] Convert labels to numpy array 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 384ee6b..2c66774 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -206,7 +206,7 @@ def _ssd_test(args: argparse.Namespace) -> None: for filename in files: with open(filename, "rb") as file: # get labels per batch - _labels = pickle.load(file) + _labels = np.asarray(pickle.load(file)) # exclude padded label entries real_labels = np.nonzero(_labels[:, :, 0] != -1) labels.extend(_labels[_labels[:, :, 0] != -1])