search

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.
https://spikingjelly.readthedocs.io
Other
1.35k stars 239 forks source link

如何应用SNN在格式为三维数组的数据集上? #139

Closed XianWoo closed 2 years ago

XianWoo commented 2 years ago

您好, 我使用的数据集格式如下: image

SpikingJelly给的案例均为图片数据集, 因此想请问如何替换为三维数组的数据集? 报错信息如下 image

fangwei123456 commented 2 years ago

积分后的DVS数据就是三维的，shape=[T, C, H, W]，可以参考

https://spikingjelly.readthedocs.io/zh_CN/latest/clock_driven/14_classify_dvsg.html