From 71c1b1f197e4820e6fe82888d389751fe883ea67 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 10:38:44 +0200 Subject: [PATCH] Background boxes are no longer removed before building observations Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index be400c4..14354d9 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -269,9 +269,6 @@ def _get_observations(detections: Sequence[Sequence[np.ndarray]]) -> List[List[n # iterate over images for i in range(batch_size): detections_image = np.asarray(detections[i]) - class_ids = np.argmax(detections_image[:, :-12], - axis=-1) - detections_image = detections_image[np.nonzero(class_ids[:])] overlaps = bounding_box_utils.iou(detections_image[:, -12:-8], detections_image[:, -12:-8], mode="outer_product",