def _get_cub200_dataset(root, train_transformation, eval_transformation): train_set = CUB200(root, train=True) test_set = CUB200(root, train=False) return train_eval_avalanche_datasets(train_set, test_set, train_transformation, eval_transformation)
def _get_fmnist_dataset(train_transformation, eval_transformation): train_set = FashionMNIST(expanduser("~") + "/.avalanche/data/fashionmnist/", train=True, download=True) test_set = FashionMNIST(expanduser("~") + "/.avalanche/data/fashionmnist/", train=False, download=True) return train_eval_avalanche_datasets( train_set, test_set, train_transformation, eval_transformation)
def _get_imagenet_dataset(root, train_transformation, test_transformation): train_set = ImageNet(root, split="train") test_set = ImageNet(root, split="val") return train_eval_avalanche_datasets(train_set, test_set, train_transformation, test_transformation)
def _get_tiny_imagenet_dataset(train_transformation, test_transformation): train_set = TinyImagenet(train=True) test_set = TinyImagenet(train=False) return train_eval_avalanche_datasets(train_set, test_set, train_transformation, test_transformation)
def _get_cifar10_dataset(train_transformation, eval_transformation): train_set = CIFAR10(expanduser("~") + "/.avalanche/data/cifar10/", train=True, download=True) test_set = CIFAR10(expanduser("~") + "/.avalanche/data/cifar10/", train=False, download=True) return train_eval_avalanche_datasets( train_set, test_set, train_transformation, eval_transformation)
def _get_omniglot_dataset(train_transformation, test_transform): train = Omniglot(root=expanduser("~") + "/.avalanche/data/omniglot/", train=True, download=True) test = Omniglot(root=expanduser("~") + "/.avalanche/data/omniglot/", train=False, download=True) return train_eval_avalanche_datasets( train_dataset=train, test_dataset=test, train_transformation=train_transformation, eval_transformation=test_transform)
def _get_mnist_dataset(train_transformation, test_transformation): train_set = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/", train=True, download=True) test_set = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/", train=False, download=True) return train_eval_avalanche_datasets(train_set, test_set, train_transformation, test_transformation)
def _get_mnist_dataset(): train_set = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/", train=True, download=True) test_set = MNIST(root=expanduser("~") + "/.avalanche/data/mnist/", train=False, download=True) return train_eval_avalanche_datasets( train_set, test_set, None, None, dataset_type=AvalancheDatasetType.CLASSIFICATION)