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

Attempted fix for padding

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-13 11:31:57 +02:00
parent c77d7bc157
commit 85712b3d13
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -279,15 +279,18 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
final_labels.append(labels) final_labels.append(labels)
empty_label = [0, 0, 0, 0, 0] empty_label = [0, 0, 0, 0, 0]
real_final_labels = []
for labels in final_labels: for labels in final_labels:
_labels = labels[:]
len_labels = len(labels) len_labels = len(labels)
if len_labels < max_nr_labels: if len_labels < max_nr_labels:
labels += empty_label * (max_nr_labels - len_labels) _labels.extend(empty_label * (max_nr_labels - len_labels))
real_final_labels.append(_labels)
length_dataset = len(final_image_paths) length_dataset = len(final_image_paths)
path_dataset = tf.data.Dataset.from_tensor_slices(final_image_paths) path_dataset = tf.data.Dataset.from_tensor_slices(final_image_paths)
label_dataset = tf.data.Dataset.from_tensor_slices(final_labels) label_dataset = tf.data.Dataset.from_tensor_slices(real_final_labels)
dataset = tf.data.Dataset.zip((path_dataset, label_dataset)) dataset = tf.data.Dataset.zip((path_dataset, label_dataset))
dataset = dataset.repeat(num_epochs) dataset = dataset.repeat(num_epochs)
dataset = dataset.batch(batch_size=batch_size) dataset = dataset.batch(batch_size=batch_size)