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

Changed internal coordinates to corners

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-10 17:28:46 +02:00
parent a685bba4fa
commit 25de799d80
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

@ -187,7 +187,8 @@ def get_model(use_dropout: bool,
dropout_rate=dropout_rate, dropout_rate=dropout_rate,
top_k=top_k, top_k=top_k,
scales=scales, scales=scales,
return_predictor_sizes=True return_predictor_sizes=True,
coords="corners"
) )
else: else:
model, predictor_sizes = vanilla_model( model, predictor_sizes = vanilla_model(
@ -197,7 +198,8 @@ def get_model(use_dropout: bool,
iou_threshold=iou_threshold, iou_threshold=iou_threshold,
top_k=top_k, top_k=top_k,
scales=scales, scales=scales,
return_predictor_sizes=True return_predictor_sizes=True,
coords="corners"
) )
if mode == "training": if mode == "training":