From af62d0492cace729c49f60f0d4b6f8f80bb1bcef Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 11 Jul 2019 15:03:39 +0200 Subject: [PATCH] Fixed call of save function Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index f191a11..d15200e 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -269,7 +269,7 @@ def _predict_loop(generator: Generator, use_dropout: bool, steps_per_epoch: int, transformed_predictions = transform_func(predictions, inverse_transforms) save_func(transformed_predictions, original_labels, filenames, - batch_counter) + batch_nr=batch_counter) batch_counter += 1