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
|
||||
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user