From 167899b2f887cc6ef9dec6c4b2d4590b2d385030 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 20 Mar 2019 14:55:32 +0100 Subject: [PATCH] Added iou threshold and top_k parameters to function calls Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 4013e16..7e64332 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -53,7 +53,8 @@ class SSD: """ def __init__(self, mode: str, weights_path: Optional[str] = None) -> None: - self._model = keras_ssd300.ssd_300(image_size=IMAGE_SIZE, n_classes=N_CLASSES, mode=mode) + self._model = keras_ssd300.ssd_300(image_size=IMAGE_SIZE, n_classes=N_CLASSES, + mode=mode, iou_threshold=IOU_THRESHOLD, top_k=TOP_K) self.mode = mode # load existing weights @@ -75,7 +76,8 @@ class DropoutSSD: def __init__(self, mode: str, weights_path: Optional[str] = None) -> None: self._model = keras_ssd300_dropout.ssd_300_dropout(image_size=IMAGE_SIZE, n_classes=N_CLASSES, - dropout_rate=DROPOUT_RATE, mode=mode) + dropout_rate=DROPOUT_RATE, mode=mode, + iou_threshold=IOU_THRESHOLD, top_k=TOP_K) self.mode = mode # load existing weights