add test handler

data/dataset.py

@@ -128,7 +128,7 @@ class EpisodicDataset(Dataset):
 
 @DATASET.register_module()
 class SingleFolderDataset(Dataset):
-    def __init__(self, root, pipeline):
+    def __init__(self, root, pipeline, with_path=False):
         assert os.path.isdir(root)
         self.root = root
         samples = []
@@ -139,12 +139,16 @@ class SingleFolderDataset(Dataset):
                     samples.append(path)
         self.samples = samples
         self.pipeline = transform_pipeline(pipeline)
+        self.with_path = with_path
 
     def __len__(self):
         return len(self.samples)
 
     def __getitem__(self, idx):
-        return self.pipeline(self.samples[idx])
+        if not self.with_path:
+            return self.pipeline(self.samples[idx])
+        else:
+            return self.pipeline(self.samples[idx]), self.samples[idx]
 
     def __repr__(self):
         return f"<SingleFolderDataset root={self.root} len={len(self)}>"