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")