From 91ebf9032cfd766e8cd1f7d8af1252436ffe06d0 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 30 May 2019 14:49:53 +0200 Subject: [PATCH] Provided image height and width for decode detections Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index a15aa2a..d01febf 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -203,7 +203,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset, observations = np.asarray(_get_observations(detections)) del detections - observations = ssd_output_decoder.decode_detections_fast(observations) + observations = ssd_output_decoder.decode_detections_fast(observations, + img_height=IMAGE_SIZE[0], + img_width=IMAGE_SIZE[1]) result_transformed = [] for i in range(batch_size): # apply inverse transformations to predicted bounding box coordinates