wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-06 22:51:18 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Yousef Rafat 2026-05-18 09:56:59 +03:00
parent 8f62c95f68
commit b74c6a3f2c
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy_extras/nodes_mesh_postprocess.py
View File

@ -152,6 +152,9 @@ class PaintMesh(IO.ComfyNode):
out_mesh = pack_variable_mesh_batch(out_verts, out_faces, out_colors) out_mesh = pack_variable_mesh_batch(out_verts, out_faces, out_colors)
return IO.NodeOutput(out_mesh) return IO.NodeOutput(out_mesh)
if coords.shape[-1] == 4:
coords = coords[:, 1:]
out_mesh = paint_mesh_with_voxels(mesh, coords, colors, resolution=resolution) out_mesh = paint_mesh_with_voxels(mesh, coords, colors, resolution=resolution)
return IO.NodeOutput(out_mesh) return IO.NodeOutput(out_mesh)
@ -654,6 +657,17 @@ def simplify_fn_fast(vertices, faces, colors=None, target=100000, max_edge_lengt
) )
return final_v, final_f, final_c return final_v, final_f, final_c
def compute_vertex_normals(vertices, faces):
v0 = vertices[faces[:, 0]]
v1 = vertices[faces[:, 1]]
v2 = vertices[faces[:, 2]]
face_normals = torch.cross(v1 - v0, v2 - v0, dim=-1)
n = torch.zeros_like(vertices)
for i in range(3):
n.index_add_(0, faces[:, i], face_normals)
return torch.nn.functional.normalize(n, dim=-1)
class PostProcessMesh(IO.ComfyNode): class PostProcessMesh(IO.ComfyNode):
@classmethod @classmethod