From d77c27727fbb25c245d7e6093844fbcc3d41967b Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 11 Jul 2019 17:34:26 +0200 Subject: [PATCH] Fixed transformation of predictions (2nd) 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 dfb8ced..4598381 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -302,7 +302,7 @@ def _transform_predictions(predictions: np.ndarray, inverse_transforms: Sequence y_pred=predictions, img_width=image_size, img_height=image_size, - coords="corners" + input_coords="corners" ) transformed_predictions = inverse_transform_func(decoded_predictions, inverse_transforms)