From 10e96807273bdca583b8389cd52469069e95fb4f Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 14 May 2019 11:42:38 +0200 Subject: [PATCH] Piped through forward passes from main function Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 690e0b9..2da475c 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -183,6 +183,7 @@ def _ssd_val(args: argparse.Namespace) -> None: batch_size = 16 image_size = 300 + forward_passes_per_image = 42 use_dropout = False if args.network == "ssd" else True weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5" @@ -205,9 +206,11 @@ def _ssd_val(args: argparse.Namespace) -> None: ) if args.debug: with use_summary_writer.as_default(): - ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits) + ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits, + forward_passes_per_image=forward_passes_per_image) else: - ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits) + ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits, + forward_passes_per_image=forward_passes_per_image) def _auto_encoder_val(args: argparse.Namespace) -> None: