This commit is contained in:
J.Y. authored and GitHub committed 2024-09-24 18:01:43 -07:00
1 parent e6941aca47
commit 01d3f73e2c
1 file changed
+2 -1
+2 -1
View File
@@ -577,7 +577,8 @@ def auto_orient_and_center_poses(
oriented_poses = transform @ poses
if oriented_poses.mean(dim=0)[2, 1] < 0:
oriented_poses[:, 1:3] = -1 * oriented_poses[:, 1:3]
oriented_poses[1:3, :] = -1 * oriented_poses[1:3, :]
transform[1:3, :] = -1 * transform[1:3, :]
elif method in ("up", "vertical"):
up = torch.mean(poses[:, :3, 1], dim=0)
up = up / torch.linalg.norm(up)