Fix potential bug when composing pipeline (#1466)

a0b1abab · ChaimZhu · GitHub · aad3339b · a0b1abab
Unverified Commit a0b1abab authored May 18, 2022 by ChaimZhu Committed by GitHub May 18, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

mmdet3d/datasets/pipelines/compose.py mmdet3d/datasets/pipelines/compose.py +2 -1

No files found.
--- a/mmdet3d/datasets/pipelines/compose.py
+++ b/mmdet3d/datasets/pipelines/compose.py
@@ -24,7 +24,8 @@ class Compose:
        self.transforms = []
        for transform in transforms:
            if isinstance(transform, dict):
-                if transform['type'] in PIPELINES._module_dict.keys():
+                _, key = PIPELINES.split_scope_key(transform['type'])
+                if key in PIPELINES._module_dict.keys():
                    transform = build_from_cfg(transform, PIPELINES)
                else:
                    transform = build_from_cfg(transform, MMDET_PIPELINES)