20 lines
633 B
Python
20 lines
633 B
Python
from PIL import Image
|
|
|
|
from hy3dgen.rembg import BackgroundRemover
|
|
from hy3dgen.shapegen import Hunyuan3DDiTFlowMatchingPipeline
|
|
from hy3dgen.texgen import Hunyuan3DPaintPipeline
|
|
|
|
model_path = 'tencent/Hunyuan3D-2'
|
|
pipeline_shapegen = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(model_path)
|
|
pipeline_texgen = Hunyuan3DPaintPipeline.from_pretrained(model_path)
|
|
|
|
image_path = 'assets/demo.png'
|
|
image = Image.open(image_path).convert("RGBA")
|
|
if image.mode == 'RGB':
|
|
rembg = BackgroundRemover()
|
|
image = rembg(image)
|
|
|
|
mesh = pipeline_shapegen(image=image)[0]
|
|
mesh = pipeline_texgen(mesh, image=image)
|
|
mesh.export('demo.glb')
|