octree_depth all 9

inference_triposg.py

@@ -73,7 +73,7 @@ def run_triposg(
 
     if faces > 0:
         print(f"[DEBUG] Simplifying mesh to {faces} faces")
-        mesh = simplify_mesh(mesh, faces)
+        # mesh = simplify_mesh(mesh, faces)
 
     return mesh
 

scripts/inference_triposg_scribble.py

@@ -37,7 +37,7 @@ def run_triposg_scribble(
         guidance_scale=0, # this is a CFG-distilled model
         attention_kwargs={"cross_attention_scale": prompt_confidence, "cross_attention_2_scale": scribble_confidence},
         use_flash_decoder=False, # there're some boundary problems when using flash decoder with this model
-        dense_octree_depth=8, hierarchical_octree_depth=8 # 256 resolution for faster inference
+        dense_octree_depth=9, hierarchical_octree_depth=9 # 256 resolution for faster inference
     ).samples[0]
     mesh = trimesh.Trimesh(outputs[0].astype(np.float32), np.ascontiguousarray(outputs[1]))    
     return mesh

scripts/inference_vae.py

@@ -52,7 +52,7 @@ if __name__ == "__main__":
             geometric_func,
             device,
             bounds=(-1.005, -1.005, -1.005, 1.005, 1.005, 1.005),
-            dense_octree_depth=8,
+            dense_octree_depth=9,
             hierarchical_octree_depth=9,
         )
         meshes = [trimesh.Trimesh(mesh_v_f[0].astype(np.float32), mesh_v_f[1]) for mesh_v_f in output]

triposg/inference_utils.py

@@ -128,7 +128,7 @@ def parallel_zoom(occupancy_grid, scale_factor):
 def hierarchical_extract_geometry(geometric_func: Callable,
                      device: torch.device,
                      bounds: Union[Tuple[float], List[float], float] = (-1.25, -1.25, -1.25, 1.25, 1.25, 1.25),
-                     dense_octree_depth: int = 8,
+                     dense_octree_depth: int = 9,
                      hierarchical_octree_depth: int = 9,
                      ):
     """

triposg/pipelines/pipeline_triposg_scribble.py

@@ -200,7 +200,7 @@ class TripoSGScribblePipeline(DiffusionPipeline, TransformerDiffusionMixin):
         callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
         callback_on_step_end_tensor_inputs: List[str] = ["latents"],
         bounds: Union[Tuple[float], List[float], float] = (-1.005, -1.005, -1.005, 1.005, 1.005, 1.005),
-        dense_octree_depth: int = 8, 
+        dense_octree_depth: int = 9, 
         hierarchical_octree_depth: int = 9,
         flash_octree_depth: int = 9,
         use_flash_decoder: bool = True,