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

Fixed weights path given to SSD predict

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-28 17:55:41 +02:00
parent aa42d831ed
commit ad11a5f6c6
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

@ -176,6 +176,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
batch_size = 16
image_size = 256
use_dropout = False
weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300.h5"
# load prepared ground truth
with open(f"{args.ground_truth_path}/photo_paths.bin", "rb") as file:
@ -192,9 +193,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, args.weights_path)
ssd.predict(scenenet_data, use_dropout, args.output_path, weights_file)
else:
ssd.predict(scenenet_data, use_dropout, args.output_path, args.weights_path)
ssd.predict(scenenet_data, use_dropout, args.output_path, weights_file)
def _auto_encoder_val(args: argparse.Namespace) -> None: