From 6eb81c45683279a35cea022f6ecec407047a5065 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 30 May 2019 14:16:43 +0200 Subject: [PATCH] Added debug print Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 1e0bda9..3c03dc2 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -265,6 +265,7 @@ def _get_observations(detections: Sequence[Sequence[np.ndarray]]) -> List[List[n # iterate over images for i in range(batch_size): + print(f"{i}th batch element") detections_image = np.asarray(detections[i]) overlaps = bounding_box_utils.iou(detections_image[:, -12:-8], detections_image[:, -12:-8],