wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-24 16:07:30 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

vertex_colors

Browse Source
This commit is contained in:
Yousef Rafat 2026-05-14 15:12:12 +03:00
parent d2b97b510c
commit f78a20271a
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_extras/nodes_trellis2.py
View File

@ -49,7 +49,7 @@ def pack_variable_mesh_batch(vertices, faces, colors=None):
color_counts = torch.tensor([c.shape[0] for c in colors], device=colors[0].device, dtype=torch.int64) color_counts = torch.tensor([c.shape[0] for c in colors], device=colors[0].device, dtype=torch.int64)
for i, c in enumerate(colors): for i, c in enumerate(colors):
packed_colors[i, :c.shape[0]] = c packed_colors[i, :c.shape[0]] = c
mesh.colors = packed_colors mesh.vertex_colors = packed_colors
mesh.color_counts = color_counts mesh.color_counts = color_counts
return mesh return mesh
@ -223,7 +223,7 @@ def paint_mesh_with_voxels(mesh, voxel_coords, voxel_colors, resolution):
final_colors = linear_colors.unsqueeze(0) final_colors = linear_colors.unsqueeze(0)
out_mesh = copy.deepcopy(mesh) out_mesh = copy.deepcopy(mesh)
out_mesh.colors = final_colors out_mesh.vertex_colors = final_colors
return out_mesh return out_mesh
@ -231,7 +231,7 @@ def paint_mesh_with_voxels(mesh, voxel_coords, voxel_colors, resolution):
def paint_mesh_default_colors(mesh): def paint_mesh_default_colors(mesh):
out_mesh = copy.copy(mesh) out_mesh = copy.copy(mesh)
vertex_count = mesh.vertices.shape[1] vertex_count = mesh.vertices.shape[1]
out_mesh.colors = mesh.vertices.new_zeros((1, vertex_count, 3)) out_mesh.vertex_colors = mesh.vertices.new_zeros((1, vertex_count, 3))
return out_mesh return out_mesh
class VaeDecodeShapeTrellis(IO.ComfyNode): class VaeDecodeShapeTrellis(IO.ComfyNode):
@ -1382,12 +1382,12 @@ class PostProcessMesh(IO.ComfyNode):
mesh.vertices = torch.stack(out_v) mesh.vertices = torch.stack(out_v)
mesh.faces = torch.stack(out_f) mesh.faces = torch.stack(out_f)
if out_c: if out_c:
mesh.colors = torch.stack(out_c) mesh.vertex_colors = torch.stack(out_c)
else: else:
mesh.vertices = out_v mesh.vertices = out_v
mesh.faces = out_f mesh.faces = out_f
if out_c: if out_c:
mesh.colors = out_c mesh.vertex_colors = out_c
else: else:
# Single Unbatched Mesh[V, 3] # Single Unbatched Mesh[V, 3]
@ -1396,7 +1396,7 @@ class PostProcessMesh(IO.ComfyNode):
mesh.vertices = v mesh.vertices = v
mesh.faces = f mesh.faces = f
if c is not None: if c is not None:
mesh.colors = c mesh.vertex_colors = c
return IO.NodeOutput(mesh) return IO.NodeOutput(mesh)