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

Prints non-zero classes

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-03 10:32:26 +02:00
parent 225d0307eb
commit 9c3e436579
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -272,8 +272,7 @@ def _get_observations(detections: Sequence[Sequence[np.ndarray]]) -> List[List[n
detections_image = np.asarray(detections[i]) detections_image = np.asarray(detections[i])
class_ids = np.argmax(detections_image[:, :-12], class_ids = np.argmax(detections_image[:, :-12],
axis=-1) axis=-1)
class_preds = detections_image[:, 0] print(np.nonzero(class_ids[:]))
print(class_preds)
detections_image = detections_image[np.nonzero(class_ids[:])] detections_image = detections_image[np.nonzero(class_ids[:])]
print(detections_image.shape) print(detections_image.shape)
overlaps = bounding_box_utils.iou(detections_image[:, -12:-8], overlaps = bounding_box_utils.iou(detections_image[:, -12:-8],