Update src/pose.py

src/pose.py

@@ -21,7 +21,6 @@ import hist_cub
 import losses
 import pose_estimation
 import spin
-import utils
 
 PE_KSP_TO_SPIN = {
     "Head": "Head",
@@ -1080,17 +1079,22 @@ def get_contacts(
     return contact
 
 
-def save_all(
+def save_mesh(
     smpl,
     smpl_output,
     save_path,
     fname,
 ):
-    utils.save_mesh_with_colors(
-        smpl_output.vertices[0].cpu().numpy(),
-        smpl.faces,
-        save_path / f"{fname}.obj",
+    mesh = trimesh.Trimesh(
+        vertices=smpl_output.vertices[0].cpu().numpy(),
+        faces=smpl.faces,
+        process=False,
     )
+    rot = trimesh.transformations.rotation_matrix(np.pi, [1, 0, 0])
+    mesh.apply_transform(rot)
+    mesh.export(save_path / f"{fname}.obj")
+    mesh.export(save_path / f"{fname}.glb")
+    mesh.export(save_path / f"{fname}.gltf")
 
 
 def eft_step(
@@ -1241,7 +1245,7 @@ def us_step(
         fist=args.fist,
     )
 
-    save_all(
+    save_mesh(
         smpl,
         smpl_output_us,
         save_path,