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

Adapted evaluation code to new file structure of saved labels

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-13 17:20:50 +02:00
parent 232e531a9d
commit b41b58930d
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/twomartens/masterthesis/cli.py
View File

@ -174,12 +174,9 @@ def _ssd_evaluate(args: argparse.Namespace) -> None:
for filename in files: for filename in files:
with open(filename, "rb") as file: with open(filename, "rb") as file:
# get labels per batch # get labels per batch
_labels = np.asarray(pickle.load(file)) label_dict = pickle.load(file)
# exclude padded label entries labels.extend(label_dict['labels'])
for i in range(_labels.shape[0]):
image_labels = _labels[i]
image_labels = image_labels[image_labels[:, 0] != -1]
labels.append(image_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)