From d38a6b7b6befa6e28162652e466af9b4b852c9cb Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 30 May 2019 14:30:20 +0200 Subject: [PATCH] Added further debug prints (2nd) Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 35cf31a..3303b85 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -191,6 +191,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset, batch_size = None for _ in range(forward_passes_per_image): result = np.array(ssd(inputs)) + print(f"shape result: {result.shape}") if batch_size is None: batch_size = result.shape[0] if detections is None: @@ -198,6 +199,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset, for i in range(batch_size): batch_item = result[i] + print(f"batch item: {batch_item.shape}") detections[i].extend(batch_item) observations = np.asarray(_get_observations(detections))