diff --git a/mmengine/dataset/base_dataset.py b/mmengine/dataset/base_dataset.py index 6f3b75174a7058dea715faa8befbdb8670d113c3..3aa56475eecba18912d63bf95cdc88eb9b6a99ba 100644 --- a/mmengine/dataset/base_dataset.py +++ b/mmengine/dataset/base_dataset.py @@ -19,13 +19,16 @@ class Compose: """Compose multiple transforms sequentially. Args: - transforms (Sequence[dict, callable]): Sequence of transform object or - config dict to be composed. + transforms (Sequence[dict, callable], optional): Sequence of transform + object or config dict to be composed. """ - def __init__(self, transforms: Sequence[Union[dict, Callable]]): + def __init__(self, transforms: Optional[Sequence[Union[dict, Callable]]]): self.transforms: List[Callable] = [] + if transforms is None: + transforms = [] + for transform in transforms: # `Compose` can be built with config dict with type and # corresponding arguments. diff --git a/tests/test_data/test_base_dataset.py b/tests/test_data/test_base_dataset.py index 12e637d5caab2ae9c8e64e4f48649ed1904438ec..e540f5276f833366ae0c54c9104eecc3321b3a56 100644 --- a/tests/test_data/test_base_dataset.py +++ b/tests/test_data/test_base_dataset.py @@ -318,6 +318,13 @@ class TestBaseDataset: with pytest.raises(TypeError): Compose([1]) + # when the input transform is None, do nothing + compose = Compose(None) + assert (compose(dict(img=self.imgs))['img'] == self.imgs).all() + + compose = Compose([]) + assert (compose(dict(img=self.imgs))['img'] == self.imgs).all() + @pytest.mark.parametrize('lazy_init', [True, False]) def test_getitem(self, lazy_init): dataset = BaseDataset(