mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-15 11:37:32 +08:00
Update geometry_types.py
This commit is contained in:
parent
46568bbc73
commit
16bcf4b856
@ -12,9 +12,16 @@ class VOXEL:
|
|||||||
|
|
||||||
|
|
||||||
class MESH:
|
class MESH:
|
||||||
def __init__(self, vertices: torch.Tensor, faces: torch.Tensor):
|
def __init__(self, vertices: torch.Tensor, faces: torch.Tensor,
|
||||||
self.vertices = vertices
|
uvs: torch.Tensor | None = None,
|
||||||
self.faces = faces
|
vertex_colors: torch.Tensor | None = None,
|
||||||
|
texture: torch.Tensor | None = None):
|
||||||
|
|
||||||
|
self.vertices = vertices # vertices: (B, N, 3)
|
||||||
|
self.faces = faces # faces: (B, M, 3)
|
||||||
|
self.uvs = uvs # uvs: (B, N, 2)
|
||||||
|
self.vertex_colors = vertex_colors # vertex_colors: (B, N, 3 or 4)
|
||||||
|
self.texture = texture # texture: (B, H, W, 3)
|
||||||
|
|
||||||
|
|
||||||
class File3D:
|
class File3D:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user