2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Further debug print

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-03 11:27:39 +02:00
parent 386ebd388c
commit 45d1a31129
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/main.py
View File

@ -209,7 +209,9 @@ def _ssd_test(args: argparse.Namespace) -> None:
_labels = np.asarray(pickle.load(file)) _labels = np.asarray(pickle.load(file))
# exclude padded label entries # exclude padded label entries
print(_labels.shape) print(_labels.shape)
labels.extend(_labels[_labels[:, :, 0] != -1]) cleaned_labels = _labels[_labels[:, :, 0] != -1]
print(cleaned_labels.shape)
labels.extend(cleaned_labels)
# store labels for later use # store labels for later use
with open(label_file, "wb") as file: with open(label_file, "wb") as file:
pickle.dump(labels, file) pickle.dump(labels, file)