Updated BPT node

This commit is contained in:
Easymode 2025-02-19 17:16:35 +00:00 committed by GitHub
parent f2cebadfa1
commit d34f251c36
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1280,6 +1280,8 @@ class Hy3DBPT:
"trimesh": ("TRIMESH",),
"enable_bpt": ("BOOLEAN", {"default": True}),
"temperature": ("FLOAT", {"default": 0.5}),
"pc_num": ("INT", {"default": 4096, "min": 1024, "max": 8192, "step": 1024}),
"verbose": ("BOOLEAN", {"default": False})
},
}
@ -1289,12 +1291,12 @@ class Hy3DBPT:
CATEGORY = "Hunyuan3DWrapper"
DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
def bpt(self, trimesh, enable_bpt, temperature):
def bpt(self, trimesh, enable_bpt, temperature, pc_num, verbose):
new_mesh = trimesh.copy()
if enable_bpt:
from .hy3dgen.shapegen.postprocessors import BptMesh
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1)
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=verbose)
mm.unload_all_models()
mm.soft_empty_cache()