xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-09 04:44:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

BPT samples added to node

Browse Source
This commit is contained in:
Easymode 2025-02-19 18:55:04 +00:00 committed by GitHub
parent 945400ea7e
commit c6bef17d4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hy3dgen/shapegen/bpt/utils.py
View File

@ -66,15 +66,14 @@ def apply_normalize(mesh):
def sample_pc(mesh, pc_num, with_normal=False, seed=1234):
def sample_pc(mesh, pc_num, with_normal=False, seed=1234, samples=50000):
mesh = apply_normalize(mesh)
if not with_normal:
points, _ = mesh.sample(pc_num, return_index=True)
return points
points, face_idx = trimesh.sample.sample_surface(mesh=mesh, count=50000, seed=seed)
points, face_idx = trimesh.sample.sample_surface(mesh=mesh, count=samples, seed=seed)
#points, face_idx = mesh.sample(50000, return_index=True)
normals = mesh.face_normals[face_idx]
pc_normal = np.concatenate([points, normals], axis=-1, dtype=np.float16)