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

Changed decoding function to Caffe-like implementation

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-08-11 14:56:59 +02:00
parent 0fbcf0d9ad
commit 8ad0d50b3b
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

@ -200,7 +200,7 @@ def predict(generator: callable,
image_size=image_size), image_size=image_size),
decode_func=functools.partial( decode_func=functools.partial(
_decode_predictions, _decode_predictions,
decode_func=ssd_output_decoder.decode_detections_fast, decode_func=ssd_output_decoder.decode_detections,
image_size=image_size, image_size=image_size,
confidence_threshold=confidence_threshold, confidence_threshold=confidence_threshold,
iou_threshold=iou_threshold iou_threshold=iou_threshold