Cityscapes¶
- class torch_uncertainty.datasets.segmentation.Cityscapes(root, split='train', mode='fine', target_type='instance', transform=None, target_transform=None, transforms=None)[source]¶
- property color_palette¶
Return the color palette of the dataset.
- decode_target(target)[source]¶
Decode target tensor to RGB tensor.
- Parameters:
target (torch.Tensor) – Target RGB tensor.
- Returns:
Decoded target.
- Return type:
Image.Image