From d8d7b516550e60c170171e5464ef52df46e35305 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 14:52:55 +0200 Subject: [PATCH] Made load_scenenet_data more compatible with clean code principles Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 4cf4003..670ebfa 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -236,7 +236,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]], instances: Sequence[Sequence[Sequence[dict]]], coco_path: str, batch_size: int, - resized_shape: Sequence[int], + image_size: int, training: bool, evaluation: bool, augment: bool, @@ -250,7 +250,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]], instances: instance data per frame per trajectory coco_path: path to the COCO data set batch_size: size of every batch - resized_shape: shape of input images to SSD + image_size: size of resized images training: True if training data is desired evaluation: True if evaluation-ready data is desired augment: True if training data should be augmented @@ -264,6 +264,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]], trajectories = zip(photo_paths, instances) final_image_paths = [] final_labels = [] + resized_shape = (image_size, image_size) from twomartens.masterthesis.ssd_keras.eval_utils import coco_utils