From 3d1ba8adbf207ce973e2c76c33dedf3b2cbdf321 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 30 May 2019 14:46:55 +0200 Subject: [PATCH] Removed debug prints during forward passes Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 8bc7ac2..a15aa2a 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -191,7 +191,6 @@ 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: @@ -199,7 +198,6 @@ 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))