增加dvs gesture数据集

fangwei123456 · Sep 10, 2020 · 78c687c · 78c687c
1 parent 312d632
commit 78c687c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spikingjelly/datasets/dvs_gesture.py b/spikingjelly/datasets/dvs_gesture.py
@@ -171,7 +171,9 @@ def __init__(self, root: str, use_frame=True, frames_num=10, split_by='number',
             # 如果root目录下存在events_root目录则认为数据集文件存在
             print(f'events data root {events_root} already exists.')
         else:
-           self.download_and_extract(root, events_root)
+            os.mkdir(events_root)
+            print(f'mkdir {events_root}')
+            self.download_and_extract(root, events_root)
 
         events_npy_root = os.path.join(root, 'events_npy')
         if os.path.exists(events_npy_root):