Merge pull request #515 from DenisKochetov/partcrafter-fix

Partcrafter fix

Author: MrForExample

Gen_3D_Modules/PartCrafter/partcrafter_src/pipelines/pipeline_partcrafter.py

@@ -337,7 +337,7 @@ def __call__(
                         max_num_expanded_coords=max_num_expanded_coords,
                         # verbose=True
                     )
-                    mesh = trimesh.Trimesh(mesh_v_f[0].astype(np.float32), mesh_v_f[1])
+                    mesh = trimesh.Trimesh(mesh_v_f[0].astype(np.float32), mesh_v_f[1], process=True)
                 except:
                     mesh_v_f = None
                     mesh = None

Gen_3D_Modules/PartCrafter/partcrafter_src/utils/render_utils.py

@@ -10,9 +10,60 @@
 from diffusers.utils.loading_utils import load_video
 import torch
 from torchvision.utils import make_grid
+import math
 
 os.environ['PYOPENGL_PLATFORM'] = 'egl'
 
+def explode_mesh(mesh, explosion_scale=0.4):
+    # ensure we have a Scene
+    if isinstance(mesh, trimesh.Trimesh):
+        scene = trimesh.Scene(mesh)
+    elif isinstance(mesh, trimesh.Scene):
+        scene = mesh
+    else:
+        raise ValueError(f"Expected Trimesh or Scene, got {type(mesh)}")
+
+    if len(scene.geometry) <= 1:
+        print("Nothing to explode")
+        return scene
+
+    # 1) collect (name, geom, world_center)
+    parts = []
+    for name, geom in scene.geometry.items():
+        # ← get(name) returns (4×4 world‐space matrix, parent_frame)
+        world_tf, _ = scene.graph.get(name)
+        pts = trimesh.transformations.transform_points(geom.vertices, world_tf)
+        center = pts.mean(axis=0)
+        parts.append((name, geom, center))
+
+    # compute global center
+    all_centers = np.stack([c for _,_,c in parts], axis=0)
+    global_center = all_centers.mean(axis=0)
+
+    exploded = trimesh.Scene()
+    for name, geom, center in parts:
+        dir_vec = center - global_center
+        norm = np.linalg.norm(dir_vec)
+        if norm < 1e-6:
+            dir_vec = np.random.randn(3)
+            dir_vec /= np.linalg.norm(dir_vec)
+        else:
+            dir_vec /= norm
+
+        offset = dir_vec * explosion_scale
+
+        # fetch the same 4×4, then bump just the translation
+        world_tf, _ = scene.graph.get(name)
+        world_tf = world_tf.copy()
+        world_tf[:3, 3] += offset
+
+        exploded.add_geometry(geom, transform=world_tf, geom_name=name)
+        print(f"[explode] {name} moved by {np.linalg.norm(offset):.4f}")
+
+    return exploded
+
+
+    
 def render(
     scene: pyrender.Scene,
     renderer: pyrender.Renderer,
@@ -123,13 +174,22 @@ def render_views_around_mesh(
         Tuple[List[Image.Image], List[Image.Image]], 
         Tuple[List[np.ndarray], List[np.ndarray]]
     ]:
+
+    meshes = []
+    scenes = []
 
     if not isinstance(mesh, (trimesh.Trimesh, trimesh.Scene)):
         raise ValueError("mesh must be a trimesh.Trimesh or trimesh.Scene object")
     if isinstance(mesh, trimesh.Trimesh):
-        mesh = trimesh.Scene(mesh)
-
-    scene = pyrender.Scene.from_trimesh_scene(mesh)
+        for i in range(num_views):
+            scenes.append(pyrender.Scene.from_trimesh_scene(trimesh.Scene(mesh)))
+    else:
+        for i in range(num_views):
+            value = math.sin(math.pi * (i - 1) / num_views)
+            scenes.append(pyrender.Scene.from_trimesh_scene(explode_mesh(mesh, 0.2 * value), 
+                                                            ambient_light=[0.02, 0.02, 0.02],
+                                                            bg_color=[0.0, 0.0, 0.0, 1.0]))
+            
     light = pyrender.DirectionalLight(
         color=np.ones(3), 
         intensity=light_intensity
@@ -149,9 +209,9 @@ def render_views_around_mesh(
     )
 
     images, depths = [], []
-    for pose in camera_poses:
+    for i, pose in enumerate(camera_poses):
         image, depth = render(
-            scene, renderer, camera, pose, light, 
+            scenes[i], renderer, camera, pose, light, 
             normalize_depth=normalize_depth,
             flags=flags,
             return_type=return_type

README.md

@@ -45,6 +45,18 @@ Your ComfyUI Root Directory\python_embeded\python.exe install.py
 ## Features:
 - For use cases please check out [Example Workflows](./_Example_Workflows/). [**Last update: 5/June/2025**]
   - **Note:** you need to put [Example Inputs Files & Folders](_Example_Workflows/_Example_Inputs_Files/) under ComfyUI Root Directory\ComfyUI\input folder before you can run the example workflow
+    
+- **PartCrafter**: [wgsxm/PartCrafter](https://github.com/wgsxm/PartCrafter)  
+  - Pipeline:
+    - Single image → 3D mesh with **part segmentation**
+    - Optional background removal (`rembg`)
+    - Output includes:
+      - Merged full mesh
+      - ZIP archive with individual part meshes 
+  - Model weights: [HuggingFace - PartCrafter](https://huggingface.co/wgsxm/PartCrafter)    
+  - [Workflow](./_Example_Workflows/PartCrafter.json)
+      
+  <video controls autoplay loop src="https://github.com/user-attachments/assets/b80bcc97-7381-4cf7-9ec6-ee48c8d58217"></video>
 
 - **Hunyuan3D_2.1**: [Tencent-Hunyuan/Hunyuan3D-2.1](https://github.com/Tencent-Hunyuan/Hunyuan3D-2.1)  
   - Updated two-stage pipeline: