2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added missing steps and aspect ratios to the label encoder

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-13 15:58:44 +02:00
parent 49a320ca12
commit cc74a8674e
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/twomartens/masterthesis/data.py
View File

@ -23,7 +23,6 @@ Functions:
load_scenenet_data(...): loads the SceneNet RGB-D data into a Tensorflow data set
prepare_scenenet_data(...): prepares the SceneNet RGB-D data and returns it in Python format
"""
import functools
from typing import Callable, List, Mapping, Tuple
from typing import Sequence
@ -304,16 +303,22 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
width=resized_shape[1])
]
generator = functools.partial(
data_generator.generate,
generator = data_generator.generate(
batch_size=batch_size,
shuffle=shuffle,
transformations=transformations,
label_encoder=ssd_input_encoder.SSDInputEncoder(
img_height=resized_shape[0],
img_width=resized_shape[1],
n_classes=len(cats_to_classes), # 80
predictor_sizes=predictor_sizes
img_height=resized_shape[0],
img_width=resized_shape[1],
n_classes=len(cats_to_classes), # 80
predictor_sizes=predictor_sizes,
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]]
),
returns={'processed_images', 'encoded_labels'},
keep_images_without_gt=False