CIFAR10H#

class torch_uncertainty.datasets.classification.CIFAR10H(root, train=None, transform=None, target_transform=None, download=False)[source]#

CIFAR-10H Dataset.

Parameters:

  • root (str | Path) – Root directory of dataset where file cifar-10h-probs.npy exists or will be saved to if download is set to True.

  • train (bool | None) – For API consistency, not used.

  • transform (Callable[..., Any] | None) – A function/transform that takes in a PIL image and returns a transformed version. E.g, transforms.RandomCrop. Defaults to None.

  • target_transform (Callable[..., Any] | None) – A function/transform that takes in the target and transforms it. Defaults to None.

  • download (bool) – If True, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again. Defaults to False.

download_h()[source]#

Download the CIFAR-10H dataset file.

Return type:

None