Update trellis/utils/postprocessing_utils.py

trellis/utils/postprocessing_utils.py

@@ -456,3 +456,53 @@ def to_glb(
     vertices = vertices @ np.array([[1, 0, 0], [0, 0, -1], [0, 1, 0]])
     mesh = trimesh.Trimesh(vertices, faces, visual=trimesh.visual.TextureVisuals(uv=uvs, image=texture))
     return mesh
+
+
+@spaces.GPU
+def export_full_quality_glb(
+    state: dict,
+    req: gr.Request,
+) -> Tuple[str, str]:
+    """
+    Export a full-quality GLB file directly from the 3D model state,
+    skipping unnecessary optimization steps.
+    """
+    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    gs, mesh, trial_id = unpack_state(state)
+    
+    # Generate full quality GLB directly
+    vertices = mesh.vertices.cpu().numpy()
+    faces = mesh.faces.cpu().numpy()
+    
+    # Just do the parametrization and texture baking
+    vertices, faces, uvs = postprocessing_utils.parametrize_mesh(vertices, faces)
+    
+    # Generate texture at full resolution
+    observations, extrinsics, intrinsics = postprocessing_utils.render_multiview(gs, resolution=2048, nviews=100)
+    masks = [np.any(observation > 0, axis=-1) for observation in observations]
+    extrinsics = [extrinsics[i].cpu().numpy() for i in range(len(extrinsics))]
+    intrinsics = [intrinsics[i].cpu().numpy() for i in range(len(intrinsics))]
+    
+    texture = postprocessing_utils.bake_texture(
+        vertices, faces, uvs,
+        observations, masks, extrinsics, intrinsics,
+        texture_size=2048,  # Maximum texture resolution
+        mode='opt',  # Use optimization mode for best quality
+        lambda_tv=0.01  # Light smoothing to prevent artifacts
+    )
+    
+    # Create the mesh with full quality settings
+    mesh = trimesh.Trimesh(
+        vertices, 
+        faces, 
+        visual=trimesh.visual.TextureVisuals(
+            uv=uvs,
+            image=Image.fromarray(texture)
+        )
+    )
+    
+    # Export
+    glb_path = os.path.join(user_dir, f"{trial_id}_full.glb")
+    mesh.export(glb_path)
+    
+    return glb_path, glb_path