如何填补空心管STL文件体素化表示中的间隙？

以下几种方法可以在不调整体素尺寸的前提下填补间隙，同时最大程度保留原物体形状：

1. 形态学闭运算填补间隙 #

通过膨胀+腐蚀的组合操作，先填充小间隙再修整边缘，避免过度变形：

import open3d as o3d
import numpy as np

# 假设已完成初始体素化，得到voxel_grid
# voxel_grid = o3d.geometry.VoxelGrid.create_from_triangle_mesh(mesh, voxel_size=xxx)

# 提取所有体素的网格坐标
voxel_coords = np.array([v.grid_index for v in voxel_grid.get_voxels()])
voxel_set = set(tuple(coord) for coord in voxel_coords)

# 第一步：膨胀操作，填充间隙
expanded_set = set()
for coord in voxel_coords:
    # 遍历3x3x3邻域
    for dx in (-1, 0, 1):
        for dy in (-1, 0, 1):
            for dz in (-1, 0, 1):
                expanded_set.add(tuple(coord + np.array([dx, dy, dz])))

# 第二步：腐蚀操作，去除膨胀带来的冗余体素
closed_coords = []
for coord in expanded_set:
    neighbor_count = 0
    # 统计3x3x3邻域内的体素数量
    for dx in (-1, 0, 1):
        for dy in (-1, 0, 1):
            for dz in (-1, 0, 1):
                if tuple(np.array(coord) + [dx, dy, dz]) in expanded_set:
                    neighbor_count += 1
    # 保留邻域体素足够多的核心区域（阈值可根据间隙大小调整）
    if neighbor_count >= 13:
        closed_coords.append(np.array(coord))

# 重建闭合后的体素网格
closed_voxel_grid = o3d.geometry.VoxelGrid()
closed_voxel_grid.voxel_size = voxel_grid.voxel_size
closed_voxel_grid.origin = voxel_grid.origin
for coord in closed_coords:
    # 复用原网格的颜色（如果不需要颜色可固定值）
    color = voxel_grid.get_voxel_color(coord) if tuple(coord) in voxel_set else [0.5, 0.5, 0.5]
    closed_voxel_grid.add_voxel(o3d.geometry.Voxel(coord, color))

2. 基于有符号距离场（SDF）的精准补全 #

利用原三角网格的距离场重新体素化，能精准贴合原始形状，自动填补间隙：

import open3d as o3d

# 加载原始STL网格
mesh = o3d.io.read_triangle_mesh("your_model.stl")
mesh.compute_vertex_normals()

# 计算有符号距离场（使用和原体素化相同的voxel_size）
sdf = o3d.geometry.SignedDistanceField.create_from_triangle_mesh(
    mesh,
    voxel_size=voxel_grid.voxel_size,
    bounds=mesh.get_axis_aligned_bounding_box()
)

# 从SDF重建体素网格，微调阈值填补间隙（阈值为0对应原表面，略负的值可填充微小间隙）
closed_voxel_grid = o3d.geometry.VoxelGrid.create_from_signed_distance_field(
    sdf,
    threshold=-0.1 * voxel_grid.voxel_size  # 可根据间隙大小调整
)

3. 边缘体素插值补全 #

针对边界处的小间隙，通过邻接体素的分布判断是否填充空位置：

import open3d as o3d
import numpy as np

# 假设已得到初始voxel_grid
voxel_set = set(tuple(v.grid_index) for v in voxel_grid.get_voxels())
new_voxels = []

# 定义6个轴向的邻接方向
directions = [(-1,0,0), (1,0,0), (0,-1,0), (0,1,0), (0,0,-1), (0,0,1)]

for voxel in voxel_grid.get_voxels():
    coord = voxel.grid_index
    # 找出当前体素周围的空邻接位置
    empty_neighbors = []
    for d in directions:
        neighbor_coord = tuple(coord + np.array(d))
        if neighbor_coord not in voxel_set:
            empty_neighbors.append(neighbor_coord)
    
    # 对每个空位置，统计其邻域内的体素数量，达到阈值则填充
    for empty_coord in empty_neighbors:
        adjacent_count = 0
        for d in directions:
            check_coord = tuple(np.array(empty_coord) + np.array(d))
            if check_coord in voxel_set:
                adjacent_count += 1
        # 当周围有至少3个体素时填充（阈值可调整）
        if adjacent_count >= 3:
            new_voxels.append(np.array(empty_coord))

# 将新体素添加到原网格
for coord in new_voxels:
    voxel_grid.add_voxel(o3d.geometry.Voxel(coord, voxel_grid.get_voxel_color(coord)))

注意事项 #

• 形态学闭运算的腐蚀阈值、SDF的阈值、插值邻接计数阈值都需要根据你的间隙大小微调
• SDF方法是最推荐的方案，因为完全基于原始网格的几何信息，补全后形状失真最小
• 插值补全适合处理局部小间隙，不适合大面积缺失的情况

内容的提问来源于stack exchange，提问作者Luis Barba