From 1d372aba59210ca8e1c3f241213f185759135e1e Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 9 Jul 2019 12:03:38 +0200 Subject: [PATCH] Use one trajectory only for prediction Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 2c5b0e9..8469034 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -341,7 +341,8 @@ def _ssd_test(args: argparse.Namespace) -> None: predictor_sizes=ssd_model.predictor_sizes, batch_size=batch_size, resized_shape=(image_size, image_size), - training=False, evaluation=True, augment=False) + training=False, evaluation=True, augment=False, + nr_trajectories=1) del file_names_photos, instances nr_digits = math.ceil(math.log10(math.ceil(length_dataset / batch_size)))