增加ALS DVS数据集

fangwei123456 · Sep 12, 2020 · b6ae047 · b6ae047
1 parent 4dc7071
commit b6ae047
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/spikingjelly/datasets/als_dvs.py b/spikingjelly/datasets/als_dvs.py
@@ -149,15 +149,15 @@ def __init__(self, root: str, train: bool, split_ratio=0.9, use_frame=True, fram
         for class_name in labels_dict.keys():
             class_dir = os.path.join(self.data_dir, class_name)
             for i in index:
-                self.file_name.append(os.path.join(class_dir, str(i).zfill(4)))
+                self.file_name.append(os.path.join(class_dir, class_name + '_' + str(i).zfill(4)))
 
     def __len__(self):
         return self.file_name.__len__()
 
     def __getitem__(self, index):
         if self.use_frame:
-            return self.get_frames_item(self.file_name[index])
+            return self.get_frames_item(self.file_name[index] + '.npz')
         else:
-            return self.get_events_item(self.file_name[index])
+            return self.get_events_item(self.file_name[index] + '.mat')