Fixed creation of final labels

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-04-28 18:50:38 +02:00
parent d12e7ec30d
commit 7a89a3cf7b

View File

@ -272,7 +272,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
continue
final_image_paths.append(image_path)
final_labels.append(labels)
final_labels.extend(labels)
length_dataset = len(final_image_paths)
labels_np = np.array(final_labels)