Refactor mesh generation in generate3d function to streamline processing

- Updated the clean_mesh function call to enable remeshing while simplifying the input parameters.
- Removed the original color extraction and scaling logic, focusing on direct mesh generation.
- Introduced temporary file creation for exporting the generated mesh, enhancing output management.

inference.py

@@ -62,26 +62,14 @@ def generate3d(model, rgb, ccm, device):
         data_config['faces'] = faces
 
     from kiui.mesh_utils import clean_mesh
-    orig_verts = data_config['verts'].squeeze().cpu().numpy()
-    # Extract per-vertex color for the original mesh
-    orig_verts_tensor = data_config['verts'].unsqueeze(0)  # shape [1, N, 3]
-    with torch.no_grad():
-        dec_verts = model.decoder(triplane_feature2, orig_verts_tensor)
-        orig_colors = model.rgbMlp(dec_verts).squeeze().detach().cpu().numpy()
-        print('orig_colors min/max BEFORE scaling:', orig_colors.min(), orig_colors.max())
-        # Comment out the scaling below if orig_colors is already in [0, 1]
-        # orig_colors = (orig_colors * 0.5 + 0.5).clip(0, 1)  # scale to [0, 1]
-        print('orig_colors min/max AFTER scaling:', orig_colors.min(), orig_colors.max())
-        orig_colors = np.clip(orig_colors, 0, 1)
-        orig_colors = np.power(orig_colors, 1/2.2)
-    verts, faces = clean_mesh(
-        orig_verts.astype(np.float32),
-        data_config['faces'].squeeze().cpu().numpy().astype(np.int32),
-        repair=True, remesh=False, remesh_size=0.005, remesh_iters=1
-    )
-    data_config['verts'] = torch.from_numpy(verts).to(device).contiguous()
-    data_config['faces'] = torch.from_numpy(faces).to(device).contiguous()
+    verts, faces = clean_mesh(data_config['verts'].squeeze().cpu().numpy().astype(np.float32), data_config['faces'].squeeze().cpu().numpy().astype(np.int32), repair = False, remesh=True, remesh_size=0.005, remesh_iters=1)
+    data_config['verts'] = torch.from_numpy(verts).cuda().contiguous()
+    data_config['faces'] = torch.from_numpy(faces).cuda().contiguous()
 
+    start_time = time.time()
+    with torch.no_grad():
+        mesh_path_glb = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
+        model.export_mesh(data_config, mesh_path_glb, tri_fea_2 = triplane_feature2)
     # # Build KDTree from original verts
     # tree = cKDTree(orig_verts)
 
@@ -92,7 +80,7 @@ def generate3d(model, rgb, ccm, device):
     # new_colors = orig_colors[idxs[:, 0]]
 
     # Create the new mesh with colors
-    mesh = trimesh.Trimesh(vertices=verts, faces=faces, vertex_colors=orig_colors)
+    # mesh = trimesh.Trimesh(vertices=verts, faces=faces, vertex_colors=orig_colors)
 
     # === Export OBJ/MTL/PNG ===
     obj_path = tempfile.NamedTemporaryFile(suffix=".obj", delete=False).name