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

Provided missing scales

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-28 19:18:14 +02:00
parent 36b11aaa5e
commit d9eb4c334f
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/twomartens/masterthesis/ssd.py
View File

@ -70,7 +70,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, iou_threshold=IOU_THRESHOLD, top_k=TOP_K)
mode=mode, iou_threshold=IOU_THRESHOLD, top_k=TOP_K,
scales=[0.07, 0.15, 0.33, 0.51, 0.69, 0.87, 1.05])
self.mode = mode
# load existing weights
@ -93,7 +94,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,
iou_threshold=IOU_THRESHOLD, top_k=TOP_K)
iou_threshold=IOU_THRESHOLD, top_k=TOP_K,
scales=[0.07, 0.15, 0.33, 0.51, 0.69, 0.87, 1.05])
self.mode = mode
# load existing weights