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

[dataset] added the option to choose the modality for the FARGO dataset

parent f1fe9f47
No related branches found
No related tags found
No related merge requests found
Pipeline #26760 passed
......@@ -37,7 +37,7 @@ class FargoDataset(Dataset):
"""
def __init__(self, original_directory=rc['bob.db.fargo.directory'],
annotation_directory=rc['bob.db.fargo.annotation_directory'],
transform=None, start_index=0):
transform=None, start_index=0, modality='rgb'):
"""Init function
Parameters
......@@ -58,7 +58,7 @@ class FargoDataset(Dataset):
id_labels = []
db = bob.db.fargo.Database(original_directory=original_directory)
objs = db.objects(purposes='train', modality='rgb')
objs = db.objects(purposes='train', modality=modality)
for o in objs:
self.data_files.append(o.make_path(directory=original_directory, extension='.png'))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment