Added debug print

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-30 14:16:43 +02:00
parent 098b0c04a6
commit 6eb81c4568
1 changed files with 1 additions and 0 deletions
--- 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],