Added none check

Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 12:33:42 +02:00
parent f440a03f29
commit c403560b99
1 changed files with 1 additions and 1 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -471,7 +471,7 @@ def _get_observations(detections: Sequence[np.ndarray]) -> List[np.ndarray]:
        used_boxes = None
        for j in range(overlaps.shape[0]):
            # check if box is already in existing observation
-            if j in used_boxes:
+            if used_boxes is not None and j in used_boxes:
                continue
            
            box_overlaps = overlaps[j]