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

Added confidence threshold to config

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-08-05 13:46:04 +02:00
parent f4982e364c
commit 37d5e32031
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/twomartens/masterthesis/config.py
View File

@ -56,6 +56,7 @@ _CONFIG_PROPS = {
"steps_per_val_epoch": (int, "20"), "steps_per_val_epoch": (int, "20"),
"ssd_forward_passes_per_image": (int, "10"), "ssd_forward_passes_per_image": (int, "10"),
"nr_classes": (int, "80"), "nr_classes": (int, "80"),
"ssd_confidence_threshold": (float, "0.5"),
"ssd_iou_threshold": (float, "0.45"), "ssd_iou_threshold": (float, "0.45"),
"ssd_top_k": (int, "200"), "ssd_top_k": (int, "200"),
"ssd_dropout_rate": (float, "0.5"), "ssd_dropout_rate": (float, "0.5"),