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

[dataset] returned the pose and identity label as int instead of string

parent db79866e
Branches
Tags
No related merge requests found
......@@ -64,8 +64,8 @@ class CasiaDataset(Dataset):
subject = int(path[-1])
cluster = path[-2]
self.data_files.append(os.path.join(root, name))
self.pose_labels.append(dir_to_pose_label[cluster])
id_labels.append(subject)
self.pose_labels.append(int(dir_to_pose_label[cluster]))
id_labels.append(int(subject))
self.id_labels = map_labels(id_labels)
......
......@@ -103,11 +103,11 @@ class MultiPIEDataset(Dataset):
for obj in objs:
temp = os.path.split(obj.path)
identity = int(temp[0].split('/')[2])
id_labels.append(identity)
id_labels.append(int(identity))
cropped_filename = os.path.join(root_dir, camera_to_pose[camera], temp[1])
cropped_filename += '.png'
self.data_files.append(cropped_filename)
self.pose_labels.append(camera_to_label[camera])
self.pose_labels.append(int(camera_to_label[camera]))
self.id_labels = map_labels(id_labels)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment