CIFAR10N#
- class torch_uncertainty.datasets.classification.CIFAR10N(root, train=True, file_arg='aggre_label', transform=None, target_transform=None, download=False)[source]#
CIFAR-10N Dataset.
- Parameters:
root (str) – Root directory of dataset where file
cifar-10h-probs.npy
exists or will be saved to if download is set toTrue
.train (bool, optional) – For API consistency, not used.
file_arg (str, optional) – The type of label noise to use. One of the following:
"aggre_label"
,"worse_label"
,"random_label1"
,"random_label2"
,"random_label3"
.transform (callable, optional) – A function/transform that takes in a PIL image and returns a transformed version. E.g,
transforms.RandomCrop
. Defaults toNone
.target_transform (callable, optional) – A function/transform that takes in the target and transforms it. Defaults to
None
.download (bool, optional) – 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 toFalse
.