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

bob/learn/pytorch/datasets/fargo.py

@@ -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'))