pylissom.datasets

Submodules

pylissom.datasets.datasets module

Extends torchvision.datasets with two common Lissom stimuli, Oriented Gaussians and “Gaussian” Faces

pylissom.datasets.datasets.get_dataset(train, args)

class pylissom.datasets.datasets.RandomDataset(length)

Bases: torch.utils.data.Dataset

Abstract Dataset representing random samples, subclasses must implement pylissom.datasets.RandomDataset._gen()

class pylissom.datasets.datasets.OrientatedGaussians(size, length, gaussians=2)

Bases: pylissom.datasets.datasets.RandomDataset

Dataset of random Oriented Gaussians samples, as used in Computional Maps in the Visual Cortex

class pylissom.datasets.datasets.ThreeDotFaces(size, length, faces=2)

Bases: pylissom.datasets.datasets.RandomDataset

Dataset of random Faces made of Three Gaussians Disks, as used in Computional Maps in the Visual Cortex

class pylissom.datasets.datasets.CKDataset(path_images='/home/hbari/data/X.npy', path_labels='/home/hbari/data/y.npy', path_subjects='/home/hbari/data/subjs.npy')

Bases: torch.utils.data.Dataset

pylissom.datasets.datasets.train_test_ck_samplers(ck_dataset, train, train_pct=0.5)

pylissom.datasets.datasets.subj_indep_train_test_samplers(subjs, pct)