update

common.py

@@ -495,13 +495,16 @@ def extract_3d_representations_v2(
     enable_delight: bool,
     req: gr.Request,
 ):
+    print("debug1")
     output_root = TMP_DIR
     user_dir = os.path.join(output_root, str(req.session_hash))
     gs_model, mesh_model = unpack_state(state, device="cpu")
+    print("debug2")
 
     filename = "sample"
     gs_path = os.path.join(user_dir, f"{filename}_gs.ply")
     gs_model.save_ply(gs_path)
+    print("debug3")
 
     # Rotate mesh and GS by 90 degrees around Z-axis.
     rot_matrix = [[0, 0, -1], [0, 1, 0], [1, 0, 0]]
@@ -512,13 +515,16 @@ def extract_3d_representations_v2(
     gs_rot = np.array(gs_add_rot) @ np.array(rot_matrix)
     pose = GaussianOperator.trans_to_quatpose(gs_rot)
     aligned_gs_path = gs_path.replace(".ply", "_aligned.ply")
+    print("debug4")
     GaussianOperator.resave_ply(
         in_ply=gs_path,
         out_ply=aligned_gs_path,
         instance_pose=pose,
     )
+    print("debug5")
     color_path = os.path.join(user_dir, "color.png")
     render_gs_api(aligned_gs_path, color_path)
+    print("debug6")
 
     mesh = trimesh.Trimesh(
         vertices=mesh_model.vertices.cpu().numpy(),
@@ -529,7 +535,7 @@ def extract_3d_representations_v2(
 
     mesh_obj_path = os.path.join(user_dir, f"{filename}.obj")
     mesh.export(mesh_obj_path)
-
+    print("debug7")
     mesh = backproject_api(
         delight_model=DELIGHT,
         imagesr_model=IMAGESR_MODEL,
@@ -539,6 +545,7 @@ def extract_3d_representations_v2(
         skip_fix_mesh=False,
         delight=enable_delight,
     )
+    print("debug8")
 
     mesh_glb_path = os.path.join(user_dir, f"{filename}.glb")
     mesh.export(mesh_glb_path)