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

Fixed filtering for Bayesian SSD 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-09-09 08:56:08 +02:00
parent 909d8f7bef
commit a91ff8cc22
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

@ -437,7 +437,7 @@ def _apply_entropy_filtering(observations: Sequence[np.ndarray],
maxima = ssd_output_decoder._greedy_nms(threshold_met, iou_threshold=iou_threshold)
maxima_output = np.zeros((maxima.shape[0], maxima.shape[1] + 1))
maxima_output[:, 0] = class_id
maxima_output[:, 1:] = single_class
maxima_output[:, 1:] = maxima
final_image_observations.append(maxima_output)
if final_image_observations:
final_image_observations = np.concatenate(final_image_observations, axis=0)