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

Fixed Bayesian NMS

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-09-08 22:57:10 +02:00
parent ec03b0df4e
commit 909d8f7bef
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -431,7 +431,7 @@ def _apply_entropy_filtering(observations: Sequence[np.ndarray],
filtered_image_observations = observations[i][observations[i][:, -1] < entropy_threshold] filtered_image_observations = observations[i][observations[i][:, -1] < entropy_threshold]
final_image_observations = [] final_image_observations = []
for class_id in range(1, nr_classes): for class_id in range(1, nr_classes):
single_class = filtered_image_observations[:, [class_id, -1, -5, -4, -3, -2]] single_class = filtered_image_observations[:, [class_id, -5, -4, -3, -2]]
threshold_met = single_class[single_class[:, 0] > confidence_threshold] threshold_met = single_class[single_class[:, 0] > confidence_threshold]
if threshold_met.shape[0] > 0: if threshold_met.shape[0] > 0:
maxima = ssd_output_decoder._greedy_nms(threshold_met, iou_threshold=iou_threshold) maxima = ssd_output_decoder._greedy_nms(threshold_met, iou_threshold=iou_threshold)