增加dvs gesture数据集

fangwei123456 · Sep 10, 2020 · 9575c3c · 9575c3c
1 parent b851821
commit 9575c3c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/spikingjelly/datasets/dvs_gesture.py b/spikingjelly/datasets/dvs_gesture.py
@@ -209,3 +209,9 @@ def __init__(self, root: str, use_frame=True, frames_num=10, split_by='number',
             self.data_dir = events_npy_root
             self.get_item_fun = DvsGesture.get_events_item
 
+
+    def __len__(self):
+        return self.file_name.__len__()
+    def __getitem__(self, index):
+        return self.get_item_fun(self.file_name[index])
+