Updated ssd_train function to use new signature of predict function

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:56:42 +02:00
parent 6c431e5e70
commit e8a82fbfaf
1 changed files with 4 additions and 3 deletions
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@ -289,12 +289,13 @@ def _ssd_test(args: argparse.Namespace) -> None:
    nr_digits = _get_nr_digits(length_dataset, batch_size)
    steps_per_epoch = _get_nr_batches(length_dataset, batch_size)
    ssd.predict(test_generator,
-                steps_per_epoch,
                ssd_model,
-                use_dropout,
-                forward_passes_per_image,
+                steps_per_epoch,
                image_size,
+                batch_size,
+                forward_passes_per_image,
                output_path,
+                use_dropout,
                nr_digits)