update gf_view_finder

2024-09-02 23:47:52 +08:00
parent 2fcfcd1966
commit e0fb9a7617
7 changed files with 78 additions and 31 deletions
--- a/utils/pose.py
+++ b/utils/pose.py
@@ -184,11 +184,11 @@ class PoseUtil:
        ], f"the rotation mode {rot_mode} is not supported!"

        if rot_mode == "quat_wxyz" or rot_mode == "quat_xyzw":
-            pose_dim = 4
+            pose_dim = 7
        elif rot_mode == "euler_xyz":
-            pose_dim = 3
-        elif rot_mode == "euler_xyz_sx_cx" or rot_mode == "rot_matrix":
            pose_dim = 6
+        elif rot_mode == "euler_xyz_sx_cx" or rot_mode == "rot_matrix":
+            pose_dim = 9
        else:
            raise NotImplementedError
        return pose_dim