1

app.py

@@ -34,10 +34,7 @@ from step1x3d_geometry.models.pipelines.pipeline import Step1X3DGeometryPipeline
 from step1x3d_texture.pipelines.step1x_3d_texture_synthesis_pipeline import (
     Step1X3DTexturePipeline,
 )
-from step1x3d_texture.utils.shape_post_process import (
-    FaceReducer,
-    DegenerateFaceRemover,
-)
+from step1x3d_geometry.models.pipelines.pipeline_utils import reduce_face, remove_degenerate_face
 
 
 parser = argparse.ArgumentParser()
@@ -88,8 +85,8 @@ def generate_func(
     geometry_mesh = out.mesh[0]
     geometry_mesh.export(geometry_save_path)
 
-    geometry_mesh = DegenerateFaceRemover()(geometry_mesh)
-    geometry_mesh = FaceReducer()(geometry_mesh)
+    geometry_mesh = remove_degenerate_face()(geometry_mesh)
+    geometry_mesh = reduce_face()(geometry_mesh)
     textured_mesh = texture_model(input_image_path, geometry_mesh)
     textured_save_path = f"{args.cache_dir}/{save_name}-textured.glb"
     textured_mesh.export(textured_save_path)

step1x3d_texture/pipelines/step1x_3d_texture_synthesis_pipeline.py

@@ -24,11 +24,7 @@ import trimesh
 import xatlas
 import scipy.sparse
 from scipy.sparse.linalg import spsolve
-from step1x3d_texture.utils.shape_post_process import (
-    FaceReducer,
-    FloaterRemover,
-    DegenerateFaceRemover,
-)
+
 from step1x3d_geometry.models.pipelines.pipeline_utils import smart_load_model
 
 
@@ -333,7 +329,7 @@ class Step1X3DTexturePipeline:
         return texture
 
     @torch.no_grad()
-    def __call__(self, image, mesh, remove_bg=True):
+    def __call__(self, image, mesh, remove_bg=True, seed=2025):
         if remove_bg:
             birefnet = AutoModelForImageSegmentation.from_pretrained(
                 "ZhengPeng7/BiRefNet", trust_remote_code=True
@@ -367,7 +363,7 @@ class Step1X3DTexturePipeline:
                 width=768,
                 num_inference_steps=self.config.num_inference_steps,
                 guidance_scale=self.config.guidance_scale,
-                seed=self.config.seed,
+                seed=seed if seed is not None else self.config.seed,
                 lora_scale=self.config.lora_scale,
                 reference_conditioning_scale=self.config.reference_conditioning_scale,
                 negative_prompt=self.config.negative_prompt,

step1x3d_texture/utils/shape_post_process.py

@@ -1,145 +0,0 @@
-# Hunyuan 3D is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT
-# except for the third-party components listed below.
-# Hunyuan 3D does not impose any additional limitations beyond what is outlined
-# in the repsective licenses of these third-party components.
-# Users must comply with all terms and conditions of original licenses of these third-party
-# components and must ensure that the usage of the third party components adheres to
-# all relevant laws and regulations.
-
-# For avoidance of doubts, Hunyuan 3D means the large language models and
-# their software and algorithms, including trained model weights, parameters (including
-# optimizer states), machine-learning model code, inference-enabling code, training-enabling code,
-# fine-tuning enabling code and other elements of the foregoing made publicly available
-# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
-
-import os
-import tempfile
-from typing import Union
-
-import numpy as np
-import pymeshlab
-import torch
-import trimesh
-
-
-def load_mesh(path):
-    if path.endswith(".glb"):
-        mesh = trimesh.load(path)
-    else:
-        mesh = pymeshlab.MeshSet()
-        mesh.load_new_mesh(path)
-    return mesh
-
-
-def reduce_face(mesh: pymeshlab.MeshSet, max_facenum: int = 200000):
-    if max_facenum > mesh.current_mesh().face_number():
-        return mesh
-
-    mesh.apply_filter(
-        "meshing_decimation_quadric_edge_collapse",
-        targetfacenum=max_facenum,
-        qualitythr=1.0,
-        preserveboundary=True,
-        boundaryweight=3,
-        preservenormal=True,
-        preservetopology=True,
-        autoclean=True,
-    )
-    return mesh
-
-
-def remove_floater(mesh: pymeshlab.MeshSet):
-    mesh.apply_filter(
-        "compute_selection_by_small_disconnected_components_per_face", nbfaceratio=0.005
-    )
-    mesh.apply_filter("compute_selection_transfer_face_to_vertex", inclusive=False)
-    mesh.apply_filter("meshing_remove_selected_vertices_and_faces")
-    return mesh
-
-
-def pymeshlab2trimesh(mesh: pymeshlab.MeshSet):
-    with tempfile.NamedTemporaryFile(suffix=".ply", delete=False) as temp_file:
-        mesh.save_current_mesh(temp_file.name)
-        mesh = trimesh.load(temp_file.name)
-    # 检查加载的对象类型
-    if isinstance(mesh, trimesh.Scene):
-        combined_mesh = trimesh.Trimesh()
-        # 如果是Scene，遍历所有的geometry并合并
-        for geom in mesh.geometry.values():
-            combined_mesh = trimesh.util.concatenate([combined_mesh, geom])
-        mesh = combined_mesh
-    return mesh
-
-
-def trimesh2pymeshlab(mesh: trimesh.Trimesh):
-    with tempfile.NamedTemporaryFile(suffix=".ply", delete=False) as temp_file:
-        if isinstance(mesh, trimesh.scene.Scene):
-            for idx, obj in enumerate(mesh.geometry.values()):
-                if idx == 0:
-                    temp_mesh = obj
-                else:
-                    temp_mesh = temp_mesh + obj
-            mesh = temp_mesh
-        mesh.export(temp_file.name)
-        mesh = pymeshlab.MeshSet()
-        mesh.load_new_mesh(temp_file.name)
-    return mesh
-
-
-def import_mesh(
-    mesh: Union[pymeshlab.MeshSet, trimesh.Trimesh, str],
-) -> pymeshlab.MeshSet:
-    if isinstance(mesh, str):
-        mesh = load_mesh(mesh)
-
-    if isinstance(mesh, (trimesh.Trimesh, trimesh.scene.Scene)):
-        mesh = trimesh2pymeshlab(mesh)
-
-    return mesh
-
-
-def export_mesh(input, output):
-    if isinstance(input, pymeshlab.MeshSet):
-        mesh = output
-    else:
-        mesh = pymeshlab2trimesh(output)
-    return mesh
-
-
-class FaceReducer:
-    def __call__(
-        self,
-        mesh: Union[pymeshlab.MeshSet, trimesh.Trimesh, str],
-        max_facenum: int = 40000,
-    ) -> Union[pymeshlab.MeshSet, trimesh.Trimesh]:
-        ms = import_mesh(mesh)
-        ms = reduce_face(ms, max_facenum=max_facenum)
-        mesh = export_mesh(mesh, ms)
-        return mesh
-
-
-class FloaterRemover:
-    def __call__(
-        self,
-        mesh: Union[pymeshlab.MeshSet, trimesh.Trimesh, str],
-    ) -> Union[pymeshlab.MeshSet, trimesh.Trimesh]:
-        ms = import_mesh(mesh)
-        ms = remove_floater(ms)
-        mesh = export_mesh(mesh, ms)
-        return mesh
-
-
-class DegenerateFaceRemover:
-    def __call__(
-        self,
-        mesh: Union[pymeshlab.MeshSet, trimesh.Trimesh, str],
-    ) -> Union[pymeshlab.MeshSet, trimesh.Trimesh]:
-        ms = import_mesh(mesh)
-
-        with tempfile.NamedTemporaryFile(suffix=".ply", delete=False) as temp_file:
-            ms.save_current_mesh(temp_file.name)
-            ms = pymeshlab.MeshSet()
-            ms.load_new_mesh(temp_file.name)
-
-        mesh = export_mesh(mesh, ms)
-        return mesh