extorch.vision.dataset

BaseDataset
CVDataset
CVClassificationDataset
CIFAR10
CIFAR100
MNIST
FashionMNIST
SVHN

class extorch.vision.dataset.BaseDataset(data_dir: str)

class extorch.vision.dataset.CVDataset(data_dir: str, train_transform: torchvision.transforms.transforms.Compose, test_transform: torchvision.transforms.transforms.Compose)

class extorch.vision.dataset.CVClassificationDataset(data_dir: str, train_transform: torchvision.transforms.transforms.Compose, test_transform: torchvision.transforms.transforms.Compose)

class extorch.vision.dataset.CIFAR10(data_dir: str, train_transform: torchvision.transforms.transforms.Compose = Compose(     RandomCrop(size=(32, 32), padding=4)     RandomHorizontalFlip(p=0.5)     ToTensor()     Normalize(mean=[0.49139968, 0.48215827, 0.44653124], std=[0.24703233, 0.24348505, 0.26158768]) ), test_transform: torchvision.transforms.transforms.Compose = Compose(     ToTensor()     Normalize(mean=[0.49139968, 0.48215827, 0.44653124], std=[0.24703233, 0.24348505, 0.26158768]) ))

class extorch.vision.dataset.CIFAR100(data_dir: str, train_transform: torchvision.transforms.transforms.Compose = Compose(     RandomCrop(size=(32, 32), padding=4)     RandomHorizontalFlip(p=0.5)     ToTensor()     Normalize(mean=[0.5070751592371322, 0.4865488733149497, 0.44091784336703466], std=[0.26733428587924063, 0.25643846291708833, 0.27615047132568393]) ), test_transform: torchvision.transforms.transforms.Compose = Compose(     ToTensor()     Normalize(mean=[0.5070751592371322, 0.4865488733149497, 0.44091784336703466], std=[0.26733428587924063, 0.25643846291708833, 0.27615047132568393]) ))

class extorch.vision.dataset.MNIST(data_dir: str, train_transform: torchvision.transforms.transforms.Compose = Compose(ToTensor()), test_transform: torchvision.transforms.transforms.Compose = Compose(ToTensor()))

class extorch.vision.dataset.FashionMNIST(data_dir: str, train_transform: torchvision.transforms.transforms.Compose = Compose(ToTensor()), test_transform: torchvision.transforms.transforms.Compose = Compose(ToTensor()))

class extorch.vision.dataset.SVHN(data_dir: str, train_transform: torchvision.transforms.transforms.Compose = Compose(     ToTensor()     Normalize(mean=[0.4377, 0.4438, 0.4728], std=[0.198, 0.201, 0.197]) ), test_transform: torchvision.transforms.transforms.Compose = Compose(     ToTensor()     Normalize(mean=[0.4377, 0.4438, 0.4728], std=[0.198, 0.201, 0.197]) ), extra_transform: torchvision.transforms.transforms.Compose = Compose(     ToTensor()     Normalize(mean=[0.4377, 0.4438, 0.4728], std=[0.198, 0.201, 0.197]) ))