ADE20k

class composer.datasets.ADE20k(datadir, split='training', both_transforms=None, image_transforms=None, target_transforms=None)

PyTorch Dataset for ADE20k.

Parameters

• datadir (str) – the path to the ADE20k folder.

• split (str) – the dataset split to use, either ‘training’, ‘validation’, or ‘test’. Default: 'training'.

• both_transforms (Module) – transformations to apply to the image and target simultaneously. Default: None.

• image_transforms (Module) – transformations to apply to the image only. Default: None.

• target_transforms (Module) – transformations to apply to the target only. Default None.