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

Fix not created directory

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-03 11:45:09 +02:00
parent 5ea412a2d0
commit 9e0edb5d79
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -181,6 +181,7 @@ 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 os
import pickle import pickle
import numpy as np import numpy as np
@ -199,6 +200,7 @@ def _ssd_test(args: argparse.Namespace) -> None:
label_file = f"{output_path}/labels.bin" label_file = f"{output_path}/labels.bin"
predictions_file = f"{output_path}/predictions.bin" predictions_file = f"{output_path}/predictions.bin"
predictions_per_class_file = f"{output_path}/predictions_class.bin" predictions_per_class_file = f"{output_path}/predictions_class.bin"
os.makedirs(evaluation_path, exist_ok=True)
# retrieve labels and un-batch them # retrieve labels and un-batch them
files = glob.glob(f"{output_path}/*ssd_labels*") files = glob.glob(f"{output_path}/*ssd_labels*")