Changed internal coordinates to corners

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

View File

@ -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":