Added debug print for class ids

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-30 14:52:20 +02:00
parent 91ebf9032c
commit bfc4adc93e
1 changed files with 1 additions and 0 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -272,6 +272,7 @@ def _get_observations(detections: Sequence[Sequence[np.ndarray]]) -> List[List[n
        detections_image = np.asarray(detections[i])
        class_ids = np.argmax(detections_image[:, :-12],
                              axis=-1)
+        print(class_ids)
        detections_image = detections_image[np.nonzero(class_ids[:])]
        print(detections_image.shape)
        overlaps = bounding_box_utils.iou(detections_image[:, -12:-8],