CamVid#
- class torch_uncertainty.datasets.segmentation.CamVid(root, group_classes=True, split=None, transforms=None, download=False)[source]#
CamVid Dataset.
- Parameters:
root (str | Path) – Root directory of dataset where
camvid/
exists or will be saved to if download is set toTrue
.group_classes (bool, optional) – Whether to group the 32 classes into 11 superclasses. Defaults to
True
.split (str, optional) – The dataset split, supports
train
,val
andtest
. Defaults toNone
.transforms (callable, optional) – A function/transform that takes input sample and its target as entry and returns a transformed version. 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
.
- property color_palette#
Return the color palette of the dataset.