build_ade20k_dataloader

composer.datasets.build_ade20k_dataloader(global_batch_size, datadir, *, split='train', drop_last=True, shuffle=True, base_size=512, min_resize_scale=0.5, max_resize_scale=2.0, final_size=512, ignore_background=True, **dataloader_kwargs)

Builds an ADE20k dataloader.

Parameters

• global_batch_size (int) – Global batch size.

• datadir (str) – Path to location of dataset.

• split (str) – The dataset split to use either ‘train’, ‘val’, or ‘test’. Default: 'train`.

• drop_last (bool) – Whether to drop last samples. Default: True.

• shuffle (bool) – Whether to shuffle the dataset. Default: True.

• base_size (int) – Initial size of the image and target before other augmentations. Default: 512.

• min_resize_scale (float) – The minimum value the samples can be rescaled. Default: 0.5.

• max_resize_scale (float) – The maximum value the samples can be rescaled. Default: 2.0.

• final_size (int) – The final size of the image and target. Default: 512.

• ignore_background (bool) – If true, ignore the background class when calculating the training loss. Default: true.

• **dataloader_kwargs (Dict[str, Any]) – Additional settings for the dataloader (e.g. num_workers, etc.)