2martens
/
masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added none check 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-08-29 12:33:42 +02:00
parent f440a03f29
commit c403560b99
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/ssd.py
View File

@ -471,7 +471,7 @@ def _get_observations(detections: Sequence[np.ndarray]) -> List[np.ndarray]:
used_boxes = None used_boxes = None
for j in range(overlaps.shape[0]): for j in range(overlaps.shape[0]):
# check if box is already in existing observation # 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 continue
box_overlaps = overlaps[j] box_overlaps = overlaps[j]