Add explicit error logging + last-resort MC fallback in flash_extract_geometry

patches/triposg/triposg/inference_utils.py

@@ -481,6 +481,7 @@ def flash_extract_geometry(
     try:
         print("final grids shape = ", grid_logits.shape)
         if DiffDMC is not None:
+            print("[TripoSG] Attempting DiffDMC extraction...")
             dmc = DiffDMC(dtype=torch.float32).to(grid_logits.device)
             sdf = -grid_logits / octree_resolution
             sdf = sdf.to(torch.float32).contiguous()
@@ -489,6 +490,7 @@ def flash_extract_geometry(
             faces = faces.detach().cpu().numpy()[:, ::-1]
             vertices = vertices / (2 ** octree_depth) * bbox_size + bbox_min
             mesh_v_f = (vertices.astype(np.float32), np.ascontiguousarray(faces))
+            print(f"[TripoSG] DiffDMC succeeded: {len(vertices)} vertices, {len(faces)} faces")
         else:
             # DiffDMC unavailable — fall back to marching cubes
             print("[TripoSG] DiffDMC unavailable, falling back to marching cubes")
@@ -497,9 +499,21 @@ def flash_extract_geometry(
             vertices, faces, normals, _ = measure.marching_cubes(gl_clean, 0, method="lewiner")
             vertices = vertices / (2 ** octree_depth) * bbox_size + bbox_min
             mesh_v_f = (vertices.astype(np.float32), np.ascontiguousarray(faces))
+            print(f"[TripoSG] marching cubes succeeded: {len(vertices)} vertices")
     except Exception as e:
-        print(e)
+        import traceback as _tb
+        print(f"[TripoSG] mesh extraction failed: {e}")
+        print(_tb.format_exc())
         torch.cuda.empty_cache()
-        mesh_v_f = (None, None)
+        # Last-resort: marching cubes on nan-cleaned grid
+        try:
+            gl_np = np.nan_to_num(grid_logits.float().cpu().numpy(), nan=0.0)
+            vertices, faces, normals, _ = measure.marching_cubes(gl_np, 0, method="lewiner")
+            vertices = vertices / max(gl_np.shape) * bbox_size + bbox_min
+            mesh_v_f = (vertices.astype(np.float32), np.ascontiguousarray(faces))
+            print(f"[TripoSG] last-resort MC succeeded: {len(vertices)} vertices")
+        except Exception as e2:
+            print(f"[TripoSG] last-resort MC also failed: {e2}")
+            mesh_v_f = (None, None)
 
     return [mesh_v_f]