Spikevision Datasets: SHD Dataset
Spiking Heidelberg Digits Dataset DataLoader
Example:
import snntorch as snn
from snntorch.spikevision import spikedata
from torch.utils.data import DataLoader
# create datasets
train_ds = spikedata.SHD("dataset/shd", train=True)
test_ds = spikedata.SHD("dataset/shd", train=False)
# create dataloaders
train_dl = DataLoader(train_ds, shuffle=True, batch_size=64)
test_dl = DataLoader(test_ds, shuffle=False, batch_size=64)
Visualizing the data:
import matplotlib.pyplot as plt
import snntorch.spikeplot as splt
# choose a random sample
n = 6295
# initialize figure and axes
fig = plt.figure(facecolor="w", figsize=(10, 5))
ax = fig.add_subplot(111)
# use spikeplot to generate a raster
splt.raster(train_dl.dataset[n][0], ax, s=1.5, c="black")