Register Compose to TRANSFORMS (#2600)

a019d2a3 · Mashiro · GitHub · 834f94db · a019d2a3
Unverified Commit a019d2a3 authored Feb 17, 2023 by Mashiro Committed by GitHub Feb 17, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

mmcv/transforms/wrappers.py mmcv/transforms/wrappers.py +1 -0

No files found.
--- a/mmcv/transforms/wrappers.py
+++ b/mmcv/transforms/wrappers.py
@@ -36,6 +36,7 @@ except ImportError:
            pass


+@TRANSFORMS.register_module()
 class Compose(BaseTransform):
    """Compose multiple transforms sequentially.