From 2d7a8e915f9e118036eb88fc8b82bed2ec0abb12 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 15 May 2019 14:53:10 +0200 Subject: [PATCH] Adapted main to different storage situation of labels 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 ab07473..256589d 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -179,7 +179,7 @@ def _ssd_test(args: argparse.Namespace) -> None: for filename in files: with open(filename, "rb") as file: # get labels per batch - _labels = np.load(file, allow_pickle=False, fix_imports=False) + _labels = pickle.load(file) # exclude padded label entries real_labels = _labels[:, :, 0] != -1 labels.extend(_labels[real_labels])