vlkit.datasets package

Submodules

vlkit.datasets.datasets module

class vlkit.datasets.datasets.CACDDataset(root, filelist)

Bases: torch.utils.data.dataset.Dataset

class vlkit.datasets.datasets.FileListDataset(filelist, root=None, transform=None, target2indice=False)

Bases: torch.utils.data.dataset.Dataset

vlkit.datasets.datasets.cifar10(path='data/cifar10', bs=100, num_workers=8)

vlkit.datasets.datasets.cifar100(path='data/cifar100', bs=256, num_workers=8)

vlkit.datasets.datasets.ilsvrc2012(path, bs=256, num_workers=8, crop_size=224)

vlkit.datasets.datasets.svhn(path='data/svhn', bs=100, num_workers=8)

vlkit.datasets.filelist module

class vlkit.datasets.filelist.FileListDataset(filelist, root_dir=None, transform=None)

Bases: torch.utils.data.dataset.Dataset

property num_classes

Module contents