From a97b9d8d216e163b1bbfb7a6172c39af59e42241 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 17:12:15 +0200 Subject: [PATCH] Set internal coordinates to "corners" Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 094e265..09149aa 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -341,6 +341,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]], n_classes=len(cats_to_classes), # 80 predictor_sizes=predictor_sizes, steps=[8, 16, 32, 64, 100, 300], + coords="corners", aspect_ratios_per_layer=[[1.0, 2.0, 0.5], [1.0, 2.0, 0.5, 3.0, 1.0 / 3.0], [1.0, 2.0, 0.5, 3.0, 1.0 / 3.0],