Fix a bug:CenterShift needs to apply on extrinsic too

HaoyiZhu · HaoyiZhu · commit 82ad88a2afca · 2024-04-09T17:50:34.000+08:00
diff --git a/configs/s3dis/pretrain-ponder-spunet-v1m1-0-base.py b/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,
diff --git a/configs/scannet/pretrain-ponder-ppt-v1m1-0-sc-s3-st-spunet.py b/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,
diff --git a/configs/scannet/pretrain-ponder-spunet-v1m1-0-base.py b/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,
diff --git a/configs/structured3d/pretrain-ponder-spunet-v1m1-0-base.py b/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,
