From c893954120dccf6a43158f141fd4bc75177ddb8e Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 5 Jun 2019 12:56:07 +0200 Subject: [PATCH] Added aspect_ratios_per_layer to arguments for SSDInputEncoder Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 7f11d6b..dc0ab41 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -404,7 +404,13 @@ def train(dataset: tf.data.Dataset, # input encoder input_encoder = ssd_input_encoder.SSDInputEncoder(IMAGE_SIZE[0], IMAGE_SIZE[1], N_CLASSES, ssd.predictor_sizes, - steps=[8, 16, 32, 64, 100, 300]) + steps=[8, 16, 32, 64, 100, 300], + 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], + [1.0, 2.0, 0.5, 3.0, 1.0 / 3.0], + [1.0, 2.0, 0.5], + [1.0, 2.0, 0.5]]) def _get_last_epoch(epoch_var: tf.Variable, **kwargs) -> int: return int(epoch_var)