Fixed creation of final labels
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -272,7 +272,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
|
|||||||
continue
|
continue
|
||||||
|
|
||||||
final_image_paths.append(image_path)
|
final_image_paths.append(image_path)
|
||||||
final_labels.append(labels)
|
final_labels.extend(labels)
|
||||||
|
|
||||||
length_dataset = len(final_image_paths)
|
length_dataset = len(final_image_paths)
|
||||||
labels_np = np.array(final_labels)
|
labels_np = np.array(final_labels)
|
||||||
|
|||||||
Reference in New Issue
Block a user