From 8ad0d50b3b5c09ef1bebd430985e55a730dea9f4 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 11 Aug 2019 14:56:59 +0200 Subject: [PATCH] Changed decoding function to Caffe-like implementation Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index d721bb2..55996dd 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -200,7 +200,7 @@ def predict(generator: callable, image_size=image_size), decode_func=functools.partial( _decode_predictions, - decode_func=ssd_output_decoder.decode_detections_fast, + decode_func=ssd_output_decoder.decode_detections, image_size=image_size, confidence_threshold=confidence_threshold, iou_threshold=iou_threshold