Class ConcatDataset

detrsmpl/data/datasets/dataset_wrappers.py:8–18

A wrapper of concatenated dataset. Same as :obj:`torch.utils.data.dataset.ConcatDataset`, but add `get_cat_ids` function. Args: datasets (list[:obj:`Dataset`]): A list of datasets.

@DATASETS.register_module()
class ConcatDataset(_ConcatDataset):
    """A wrapper of concatenated dataset.

    Same as :obj:`torch.utils.data.dataset.ConcatDataset`, but
    add `get_cat_ids` function.

    Args:
        datasets (list[:obj:`Dataset`]): A list of datasets.
    """
    def __init__(self, datasets: list):
        super(ConcatDataset, self).__init__(datasets)


@DATASETS.register_module()