Refactor color scaling and assignment in generate3d function

- Added debug prints to monitor the min/max values of orig_colors before and after scaling.
- Commented out the scaling operation to allow for flexibility if orig_colors are already in the [0, 1] range.
- Updated color assignment to use only the nearest neighbor for improved accuracy in color mapping.

inference.py

@@ -68,7 +68,10 @@ def generate3d(model, rgb, ccm, device):
     with torch.no_grad():
         dec_verts = model.decoder(triplane_feature2, orig_verts_tensor)
         orig_colors = model.rgbMlp(dec_verts).squeeze().detach().cpu().numpy()
-        orig_colors = (orig_colors * 0.5 + 0.5).clip(0, 1)  # scale to [0, 1]
+        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())
     verts, faces = clean_mesh(
         orig_verts.astype(np.float32),
         data_config['faces'].squeeze().cpu().numpy().astype(np.int32),
@@ -83,7 +86,8 @@ def generate3d(model, rgb, ccm, device):
     # For each new vertex, find the nearest old vertex and copy its color
     k = 3
     dists, idxs = tree.query(verts, k=k)
-    new_colors = np.mean(orig_colors[idxs], axis=1)
+    # Use only the nearest neighbor for color assignment
+    new_colors = orig_colors[idxs[:, 0]]
 
     # Create the new mesh with colors
     mesh = trimesh.Trimesh(vertices=verts, faces=faces, vertex_colors=new_colors)