Register Compose to TRANSFORMS (open-mmlab#2600)

tyomj · May 8, 2023 · 884ef8d · 884ef8d
1 parent 2c54424
commit 884ef8d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmcv/transforms/wrappers.py b/mmcv/transforms/wrappers.py
@@ -36,6 +36,7 @@ def nullcontext(resource=None):
             pass
 
 
+@TRANSFORMS.register_module()
 class Compose(BaseTransform):
     """Compose multiple transforms sequentially.