From 386ebd388c642e99b5aee9deec321df29efd4e0d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 11:26:02 +0200 Subject: [PATCH] Added debug print for label shape Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 87eb1b5..9e139e6 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -208,6 +208,7 @@ def _ssd_test(args: argparse.Namespace) -> None: # get labels per batch _labels = np.asarray(pickle.load(file)) # exclude padded label entries + print(_labels.shape) labels.extend(_labels[_labels[:, :, 0] != -1]) # store labels for later use with open(label_file, "wb") as file: