常用的点云预处理算法_点云处理算法

点云预处理是处理点云数据时的重要部分，其目的是提高点云数据的质量和处理效率。通过去除离群点、减少点云密度和增强特征，可以消除噪声、减少计算量、提高算法的准确性和鲁棒性，从而为后续的点云处理和分析步骤（如配准、分割和重建）提供更高质量的数据基础。

以下是常用的点云预处理算法介绍，内容包括离群点过滤、点云下采样、坐标上采样和特征上采样的数学原理以及Open3D实现代码。

离群点过滤

无效值剔除

无效值剔除是指移除点云中的值或无穷大的值。

数学原理： 假设点云数据集为，其中每个点。无效值剔除的规则可以表示为：

Open3D实现：

import open3d as o3d
 
def remove_invalid_points(pcd):
    pcd = pcd.select_by_index(
        np.where(np.isfinite(np.asarray(pcd.points)).all(axis=1))[0]
    )
    return pcd
 
# 示例
pcd = o3d.io.read_point_cloud("example.ply")
pcd = remove_invalid_points(pcd)
o3d.visualization.draw_geometries([pcd])

统计方法剔除

统计方法剔除利用每个点的邻域统计信息来识别并移除离群点。

数学原理： 计算每个点到其个最近邻点的平均距离，并将该点的平均距离与全局平均距离进行比较。如果某点的平均距离超出某一阈值，则认为该点为离群点。

Open3D实现：

def statistical_outlier_removal(pcd, nb_neighbors=20, std_ratio=2.0):
    cl, ind = pcd.remove_statistical_outlier(nb_neighbors=nb_neighbors, std_ratio=std_ratio)
    pcd = pcd.select_by_index(ind)
    return pcd
 
# 示例
pcd = o3d.io.read_point_cloud("example.ply")
pcd = statistical_outlier_removal(pcd)
o3d.visualization.draw_geometries([pcd])

半径滤波方法剔除

半径滤波方法根据每个点在指定半径内的邻居数量来剔除离群点。

数学原理： 设定一个半径和最小点数阈值。对于每个点，如果其在半径内的邻居数量少于，则认为该点为离群点。

Open3D实现：

def radius_outlier_removal(pcd, radius=0.05, min_points=5):
    cl, ind = pcd.remove_radius_outlier(nb_points=min_points, radius=radius)
    pcd = pcd.select_by_index(ind)
    return pcd
 
# 示例
pcd = o3d.io.read_point_cloud("example.ply")
pcd = radius_outlier_removal(pcd)
o3d.visualization.draw_geometries([pcd])

离群点剔除代码示例：

import open3d as o3d
from copy import deepcopy
import numpy as np
 
 
if __name__ == '__main__':
    file_path = 'rabbit.pcd'
    pcd = o3d.io.read_point_cloud(file_path)
    pcd = pcd.uniform_down_sample(50)#每50个点采样一次
    pcd.paint_uniform_color([0.5, 0.5, 0.5])#指定显示为灰色
    print(pcd)
 
    #剔除无效值
    pcd1 = deepcopy(pcd)
    pcd1.paint_uniform_color([0, 0, 1])#指定显示为蓝色
    pcd1.translate((20, 0, 0)) #整体进行x轴方向平移
    pcd1 = pcd1.remove_non_finite_points(True, True)#剔除无效值
    print(pcd1)
 
    #统计方法剔除
    pcd2 = deepcopy(pcd)
    pcd2.paint_uniform_color([0, 1, 0])#指定显示为蓝色
    pcd2.translate((-20, 0, 0)) #整体进行x轴方向平移
    res = pcd2.remove_statistical_outlier(20, 0.5)#统计方法剔除
    pcd2 = res[0]#返回点云，和点云索引
    print(pcd2)
 
    #半径方法剔除
    pcd3 = deepcopy(pcd)
    pcd3.paint_uniform_color([1, 0, 0])#指定显示为蓝色
    pcd3.translate((0, 20, 0)) #整体进行y轴方向平移
    res = pcd3.remove_radius_outlier(nb_points=20, radius=2)#半径方法剔除
    pcd3 = res[0]#返回点云，和点云索引
    print(pcd3)
 
    # # 点云显示
    o3d.visualization.draw_geometries([pcd, pcd1, pcd2, pcd3], #点云列表
                                      window_name="离群点剔除",
                                      point_show_normal=False,
                                      width=800,  # 窗口宽度
                                      height=600)  # 窗口高度
 

点云下采样

点云下采样的主要目的是减少点云数据的规模，从而降低计算复杂度和存储需求。在实际应用中，点云数据通常包含大量的点，这会占用大量的存储空间并增加处理时间。通过下采样，可以去除冗余点，只保留关键点，既能加速后续处理（如配准、分类、分割等），又能减少内存使用。此外，下采样还能提高算法的鲁棒性，减少噪声影响，使得处理结果更加稳定和可靠。

体素下采样

体素下采样通过将点云划分为固定大小的三维网格，并用每个网格中的点的质心来代表该网格中的所有点。

数学原理： 设定体素网格的大小为，将点云划分为体素网格，每个体素中的点用其质心替代。

其中，为体素的质心，为体素中的点数。

Open3D实现：

import open3d as o3d
from copy import deepcopy
 
 
if __name__ == '__main__':
    file_path = 'rabbit.pcd'
    pcd = o3d.io.read_point_cloud(file_path)
    pcd.paint_uniform_color([0.5, 0.5, 0.5])#指定显示为灰色
    print(pcd)
 
    pcd1 = deepcopy(pcd)
    pcd1.paint_uniform_color([0, 0, 1])#指定显示为蓝色
    pcd1.translate((20, 0, 0)) #整体进行x轴方向平移
    pcd1 = pcd1.voxel_down_sample(voxel_size=1)
    print(pcd1)
 
    pcd2 = deepcopy(pcd)
    pcd2.paint_uniform_color([0, 1, 0])#指定显示为绿色
    pcd2.translate((0, 20, 0)) #整体进行y轴方向平移
    res = pcd2.voxel_down_sample_and_trace(1, min_bound=pcd2.get_min_bound()-0.5, max_bound=pcd2.get_max_bound()+0.5, approximate_class=True)
    pcd2 = res[0]
    print(pcd2)
 
    
    # 点云显示
    o3d.visualization.draw_geometries([pcd, pcd1, pcd2], #点云列表
                                      window_name="体素下采样",
                                      point_show_normal=False,
                                      width=800,  # 窗口宽度
                                      height=600)  # 窗口高度
 

随机下采样

随机下采样是从点云中随机选取一定比例的点，以减少点云的点数。

数学原理： 设点云数据集为，随机下采样从中选取个点。其中，是从中随机选取的索引。

Open3D实现：

import open3d as o3d
from copy import deepcopy
import numpy as np
 
 
if __name__ == '__main__':
    file_path = 'rabbit.pcd'
    pcd = o3d.io.read_point_cloud(file_path)
    pcd.paint_uniform_color([0.5, 0.5, 0.5])#指定显示为灰色
    print(pcd)
 
    pcd1 = deepcopy(pcd)
    pcd1.paint_uniform_color([0, 0, 1])#指定显示为蓝色
    pcd1.translate((20, 0, 0)) #整体进行x轴方向平移
    pcd1 = pcd1.uniform_down_sample(100)#每100个点采样一次
    print(pcd1)
 
    pcd2 = deepcopy(pcd)
    pcd2.paint_uniform_color([0, 1, 0])#指定显示为绿色
    pcd2.translate((0, 20, 0)) #整体进行y轴方向平移
    pcd2 = pcd2.random_down_sample(0.1)#采1/10的点云
    print(pcd2)
 
    #自定义随机采样
    pcd3 = deepcopy(pcd)
    pcd3.translate((-20, 0, 0)) #整体进行x轴方向平移
    points = np.array(pcd3.points)
    n = np.random.choice(len(points), 500, replace=False) #s随机采500个数据，这种随机方式也可以自己定义
    pcd3.points = o3d.utility.Vector3dVector(points[n])
    pcd3.paint_uniform_color([1, 0, 0])#指定显示为红色
    print(pcd3)
    
    # # 点云显示
    o3d.visualization.draw_geometries([pcd, pcd1, pcd2, pcd3], #点云列表
                        

均匀下采样

均匀下采样是等间距选择点云中的点。

数学原理： 设点云数据集为，均匀下采样每隔个点选择一个点。

Open3D实现：

def uniform_downsample(pcd, every_k_points=10):
    pcd = pcd.uniform_down_sample(every_k_points)
    return pcd
 
# 示例
pcd = o3d.io.read_point_cloud("example.ply")
pcd = uniform_downsample(pcd)
o3d.visualization.draw_geometries([pcd])

最远点采样

最远点采样是一种迭代方法，每次选择距离当前已选点集最远的点。

数学原理： 初始化点集为随机选择的一个点，每次迭代选择距离当前已选点集最远的点：

Open3D实现：

import open3d as o3d
from copy import deepcopy
import numpy as np
 
def farthest_point_sample(point, npoint):
    """
    Input:
        xyz: pointcloud data, [N, D]
        npoint: number of samples
    Return:
        centroids: sampled pointcloud index, [npoint, D]
    """
    N, D = point.shape
    xyz = point[:,:3]
    centroids = np.zeros((npoint,))
    distance = np.ones((N,)) * 1e10
    farthest = np.random.randint(0, N)
    for i in range(npoint):
        centroids[i] = farthest
        centroid = xyz[farthest, :]
        dist = np.sum((xyz - centroid) ** 2, -1)
        mask = dist < distance
        distance[mask] = dist[mask]
        farthest = np.argmax(distance, -1)
    point = point[centroids.astype(np.int32)]
    return point
 
if __name__ == '__main__':
    file_path = 'rabbit.pcd'
    pcd = o3d.io.read_point_cloud(file_path)
    pcd.paint_uniform_color([0.5, 0.5, 0.5])#指定显示为灰色
    print(pcd)
 
    pcd1 = deepcopy(pcd)
    pcd1.translate((20, 0, 0)) #整体进行x轴方向平移
    points = np.array(pcd1.points)
    points = farthest_point_sample(points, 500)
    pcd1 = o3d.geometry.PointCloud()
    pcd1.points = o3d.utility.Vector3dVector(points)
    pcd1.paint_uniform_color([0, 0, 1])#指定显示为蓝色
    print(pcd1)
 
    # # 点云显示
    o3d.visualization.draw_geometries([pcd, pcd1], #点云列表
                                      window_name="最远点采样",
                                      point_show_normal=False,
                                      width=800,  # 窗口宽度
                                      height=600)  # 窗口高度
 

坐标上采样

点云上采样的目的是增加点云的密度，以提高其分辨率和细节表现。在许多应用中，特别是三维重建和表面细化时，原始点云的分辨率可能不足，导致重建结果不够精细或存在明显的缺陷。通过上采样，可以插值生成更多的点，使得点云更密集，从而更好地捕捉物体的几何细节，提升最终的三维模型质量。这对于需要高精度、高分辨率数据的任务（如精细的表面重建、细节检测等）尤为重要。

插值法

插值法是通过插值计算新增点的坐标。

数学原理： 常用的插值方法包括线性插值、双线性插值、三次样条插值等。以线性插值为例，两个点和之间新增点：

其中，为插值系数，取值范围为[0, 1]。

Open3D实现：

import open3d as o3d
from copy import deepcopy
 
if __name__ == '__main__':
    file_path = 'rabbit.pcd'
    pcd = o3d.io.read_point_cloud(file_path)
    pcd.paint_uniform_color([0.5, 0.5, 0.5])  # 指定显示为灰色
    print(pcd)
 
    pcd1 = deepcopy(pcd)
    pcd1.paint_uniform_color([0, 0, 1])  # 指定显示为蓝色
    pcd1.translate((20, 0, 0))  # 整体进行x轴方向平移
    pcd1 = pcd1.voxel_down_sample(voxel_size=1)
    print(pcd1)
 
    mesh = o3d.geometry.TriangleMesh.create_from_point_cloud_alpha_shape(pcd1, alpha=2)
    pcd2 = mesh.sample_points_poisson_disk(number_of_points=3000, init_factor=5)
    pcd2.paint_uniform_color([0, 1, 0])  # 指定显示为绿色
    pcd2.translate((-40, 0, 0))  # 整体进行x轴方向平移
    print(pcd2)
    o3d.visualization.draw_geometries([pcd, pcd1, pcd2],  # 点云列表
                                      window_name="点云上采样",
                                      point_show_normal=False,
                                      width=800,  # 窗口宽度
                                      height=600)  # 窗口高度

特征上采样

最近邻插值

最近邻插值是通过选择最近邻的特征值作为新增点的特征值。

数学原理： 设已知点云的特征为，新增点的特征值通过最近邻点的特征值确定：,其中，为距离最近的点。

Open3D实现：

from sklearn.neighbors import NearestNeighbors
 
def nearest_neighbor_interpolation(pcd, features, num_points):
    points = np.asarray(pcd.points)
    new_points = np.array([]).reshape(0, 3)
    new_features = np.array([]).reshape(0, features.shape[1])
 
    for i in range(len(points) - 1):
        for t in np.linspace(0, 1, num_points):
            new_point = points[i] + t * (points[i + 1] - points[i])
            new_points = np.vstack((new_points, new_point))
 
    nn = NearestNeighbors(n_neighbors=1).fit(points)
    distances, indices = nn.kneighbors(new_points)
    new_features = features[indices.flatten()]
 
    pcd_interpolated = o3d.geometry.PointCloud()
    pcd_interpolated.points = o3d.utility.Vector3dVector(new_points)
    return pcd_interpolated, new_features
 
# 示例
pcd = o3d.io.read_point_cloud("example.ply")
features = np.random.rand(len(pcd.points), 3)  # 假设特征是随机生成的
pcd, new_features = nearest_neighbor_interpolation(pcd, features, num_points=10)
o3d.visualization.draw_geometries([pcd])

以上内容总结自网络，如有帮助欢迎关注与转发，我们下次再见！