Skip to content
Snippets Groups Projects
Commit ebf73ad0 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[dataset] casia: rename id to label (to be more generic)

parent 6f3be1da
Branches
Tags
No related merge requests found
Pipeline #
......@@ -58,7 +58,7 @@ class CasiaWebFaceDataset(Dataset):
"""
image = bob.io.base.load(self.data_files[idx])
identity = self.id_labels[idx]
sample = {'image': image, 'id': identity}
sample = {'image': image, 'label': identity}
if self.transform:
sample = self.transform(sample)
......@@ -133,7 +133,7 @@ class CasiaDataset(Dataset):
image = bob.io.base.load(self.data_files[idx])
identity = self.id_labels[idx]
pose = self.pose_labels[idx]
sample = {'image': image, 'id': identity, 'pose': pose}
sample = {'image': image, 'label': identity, 'pose': pose}
if self.transform:
sample = self.transform(sample)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment