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

Improved output path

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-29 11:25:21 +02:00
parent 3d56d69122
commit 64a0253834
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

@ -177,6 +177,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
image_size = 300
use_dropout = False
weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5"
output_path = f"{args.output_path}/val/ssd/{args.iteration}/"
# load prepared ground truth
with open(f"{args.ground_truth_path}/photo_paths.bin", "rb") as file:
@ -193,9 +194,9 @@ def _ssd_val(args: argparse.Namespace) -> None:
)
if args.debug:
with use_summary_writer.as_default():
ssd.predict(scenenet_data, use_dropout, args.output_path, weights_file)
ssd.predict(scenenet_data, use_dropout, output_path, weights_file)
else:
ssd.predict(scenenet_data, use_dropout, args.output_path, weights_file)
ssd.predict(scenenet_data, use_dropout, output_path, weights_file)
def _auto_encoder_val(args: argparse.Namespace) -> None: