def cifar10_val(dataset_root): train_transform = amdim_transforms.TransformsC10() dataset = CIFAR10Mixed(root=dataset_root, split='val', transform=train_transform, download=True) return dataset
def cifar10(dataset_root, split: str = 'train'): assert split in ('train', 'val') dataset = CIFAR10Mixed( root=dataset_root, split=split, transform=amdim_transforms.AMDIMTrainTransformsCIFAR10(), download=True, ) return dataset
def cifar10_train(dataset_root, patch_size, patch_overlap): train_transform = amdim_transforms.TransformsC10Patches( patch_size=patch_size, patch_overlap=patch_overlap) dataset = CIFAR10Mixed(root=dataset_root, split='train', transform=train_transform, download=True) return dataset
def cifar10(dataset_root, patch_size, patch_overlap, split: str = 'train'): assert split in ('train', 'val') train_transform = amdim_transforms.TransformsC10Patches( patch_size=patch_size, patch_overlap=patch_overlap) dataset = CIFAR10Mixed( root=dataset_root, split=split, transform=train_transform, download=True, ) return dataset