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: