Fix a bug:CenterShift needs to apply on extrinsic too

Author: HaoyiZhu

configs/s3dis/pretrain-ponder-spunet-v1m1-0-base.py

@@ -185,7 +185,13 @@
         render_semantic=True,
         num_cameras=num_cameras,
         transform=[
-            dict(type="CenterShift", apply_z=True),
+            dict(
+                type="CenterShift",
+                apply_z=True,
+                keys=[
+                    "extrinsic",
+                ],
+            ),
             dict(
                 type="RandomDropout",
                 dropout_ratio=0.8,

configs/scannet/pretrain-ponder-ppt-v1m1-0-sc-s3-st-spunet.py

@@ -267,7 +267,13 @@
                 render_semantic=True,
                 num_cameras=num_cameras,
                 transform=[
-                    dict(type="CenterShift", apply_z=True),
+                    dict(
+                        type="CenterShift",
+                        apply_z=True,
+                        keys=[
+                            "extrinsic",
+                        ],
+                    ),
                     dict(
                         type="RandomDropout",
                         dropout_ratio=0.8,
@@ -366,7 +372,13 @@
                 render_semantic=True,
                 num_cameras=num_cameras,
                 transform=[
-                    dict(type="CenterShift", apply_z=True),
+                    dict(
+                        type="CenterShift",
+                        apply_z=True,
+                        keys=[
+                            "extrinsic",
+                        ],
+                    ),
                     dict(
                         type="RandomDropout",
                         dropout_ratio=0.8,
@@ -464,7 +476,13 @@
                 render_semantic=True,
                 num_cameras=num_cameras,
                 transform=[
-                    dict(type="CenterShift", apply_z=True),
+                    dict(
+                        type="CenterShift",
+                        apply_z=True,
+                        keys=[
+                            "extrinsic",
+                        ],
+                    ),
                     dict(
                         type="RandomDropout",
                         dropout_ratio=0.8,

configs/scannet/pretrain-ponder-spunet-v1m1-0-base.py

@@ -201,7 +201,13 @@
         render_semantic=True,
         num_cameras=num_cameras,
         transform=[
-            dict(type="CenterShift", apply_z=True),
+            dict(
+                type="CenterShift",
+                apply_z=True,
+                keys=[
+                    "extrinsic",
+                ],
+            ),
             dict(
                 type="RandomDropout",
                 dropout_ratio=0.8,

configs/structured3d/pretrain-ponder-spunet-v1m1-0-base.py

@@ -209,7 +209,13 @@
         render_semantic=True,
         num_cameras=num_cameras,
         transform=[
-            dict(type="CenterShift", apply_z=True),
+            dict(
+                type="CenterShift",
+                apply_z=True,
+                keys=[
+                    "extrinsic",
+                ],
+            ),
             dict(
                 type="RandomDropout",
                 dropout_ratio=0.8,