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

Fixed wrong indexing

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-03 10:48:58 +02:00
parent 71c1b1f197
commit e097643daf
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -182,7 +182,8 @@ def _test(args: argparse.Namespace) -> None:
def _ssd_test(args: argparse.Namespace) -> None: def _ssd_test(args: argparse.Namespace) -> None:
import glob import glob
import pickle import pickle
import numpy as np
import tensorflow as tf import tensorflow as tf
from twomartens.masterthesis import evaluate from twomartens.masterthesis import evaluate
@ -207,7 +208,7 @@ def _ssd_test(args: argparse.Namespace) -> None:
# get labels per batch # get labels per batch
_labels = pickle.load(file) _labels = pickle.load(file)
# exclude padded label entries # exclude padded label entries
real_labels = _labels[:, :, 0] != -1 real_labels = np.nonzero(_labels[:, :, 0] != -1)
labels.extend(_labels[real_labels]) labels.extend(_labels[real_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: