Added confidence threshold to config

Signed-off-by: Jim Martens <github@2martens.de>
2019-08-05 13:46:04 +02:00
parent f4982e364c
commit 37d5e32031
1 changed files with 1 additions and 0 deletions
--- 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"),