From 25de799d804398d4aed0d62c08812526917d687c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 17:28:46 +0200 Subject: [PATCH] Changed internal coordinates to corners 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 f0763b0..1808f69 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -187,7 +187,8 @@ def get_model(use_dropout: bool, dropout_rate=dropout_rate, top_k=top_k, scales=scales, - return_predictor_sizes=True + return_predictor_sizes=True, + coords="corners" ) else: model, predictor_sizes = vanilla_model( @@ -197,7 +198,8 @@ def get_model(use_dropout: bool, iou_threshold=iou_threshold, top_k=top_k, scales=scales, - return_predictor_sizes=True + return_predictor_sizes=True, + coords="corners" ) if mode == "training":