From 37d5e3203185ee1e2ea71981ad045b198a698ba1 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 5 Aug 2019 13:46:04 +0200 Subject: [PATCH] Added confidence threshold to config Signed-off-by: Jim Martens --- src/twomartens/masterthesis/config.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/config.py b/src/twomartens/masterthesis/config.py index 909a515..996cbeb 100644 --- a/src/twomartens/masterthesis/config.py +++ b/src/twomartens/masterthesis/config.py @@ -56,6 +56,7 @@ _CONFIG_PROPS = { "steps_per_val_epoch": (int, "20"), "ssd_forward_passes_per_image": (int, "10"), "nr_classes": (int, "80"), + "ssd_confidence_threshold": (float, "0.5"), "ssd_iou_threshold": (float, "0.45"), "ssd_top_k": (int, "200"), "ssd_dropout_rate": (float, "0.5"),