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

13
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 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 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 Callable, List, Mapping, Tuple
from typing import Sequence from typing import Sequence
@ -304,8 +303,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
width=resized_shape[1]) width=resized_shape[1])
] ]
generator = functools.partial( generator = data_generator.generate(
data_generator.generate,
batch_size=batch_size, batch_size=batch_size,
shuffle=shuffle, shuffle=shuffle,
transformations=transformations, transformations=transformations,
@ -313,7 +311,14 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
img_height=resized_shape[0], img_height=resized_shape[0],
img_width=resized_shape[1], img_width=resized_shape[1],
n_classes=len(cats_to_classes), # 80 n_classes=len(cats_to_classes), # 80
predictor_sizes=predictor_sizes 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'}, returns={'processed_images', 'encoded_labels'},
keep_images_without_gt=False keep_images_without_gt=False