Python实现点云PCA配准——粗配准

发布于:2025-08-09 ⋅ 阅读:(16) ⋅ 点赞:(0)

        本节我们来介绍PCA(主成分分析)算法进行点云配准,这是一种经典的统计降维与特征提取工具,在三维点云处理中常被用来完成“粗配准”。其核心思想是:先把两个待对齐的点云各自进行主成分分解,获得各自的“主轴”,然后只需把源点云的主轴旋转-平移到与目标云的主轴重合,即可实现初步对齐。具体流程可概括为五步:

        1. 计算质心:求出点云中所有点的几何中心。  
        2. 去质心:把每个点坐标减去质心,使点云以原点为中心。  
        3. 协方差矩阵:在中心化后的点云上计算 3×3 协方差矩阵。  
        4. 特征分解:对协方差矩阵做特征值分解,得到三个特征向量,即为点云的三个主方向(主轴)。  
        5. 轴对齐:构造旋转矩阵和平移向量,将源云的主轴依次对应到目标云的主轴,完成粗配准。

        由于 PCA 只依赖一阶矩和二阶矩,计算量小、对噪声有一定鲁棒性,因此常被用作以下场景的“第一步”:

        • 初始配准:在运行 ICP 等精细配准算法前,用 PCA 给出良好初值,可显著减少迭代次数并避免局部极小。  
        • 姿态估计:机器人或视觉系统可通过 PCA 快速估计物体或场景的大致朝向。  
        • 多视角拼接:把不同视角拍到的点云先用 PCA 粗略对齐,为后续精细拼接提供初始解。

本次使用的数据依然我们的老朋友——兔砸!显示如下:

一、PCA配准程序

from __future__ import annotations

import copy
import logging
from typing import Tuple

import numpy as np
import open3d as o3d

logging.basicConfig(level=logging.INFO, format="[%(levelname)s] %(message)s")


# ------------------------------------------------------------------------------
# 工具函数
# ------------------------------------------------------------------------------
def estimate_covariances(
    pcd: o3d.geometry.PointCloud,
    search_param: o3d.geometry.KDTreeSearchParamHybrid,
) -> Tuple[np.ndarray, np.ndarray]:
    """返回点云质心与 3×3 协方差矩阵"""
    center, cov = pcd.compute_mean_and_covariance()
    return np.asarray(center), np.asarray(cov)


def pca_registration(
    source: o3d.geometry.PointCloud,
    target: o3d.geometry.PointCloud,
) -> o3d.geometry.PointCloud:
    """
    使用 PCA 对齐两个点云的主轴,再用 Kabsch 精细对齐
    返回 : 配准后的 source 点云(拷贝)
    """
    # 1. 计算质心与主轴方向
    c_src, cov_src = estimate_covariances(
        source, o3d.geometry.KDTreeSearchParamHybrid(radius=0.01, max_nn=30)
    )
    c_tgt, cov_tgt = estimate_covariances(
        target, o3d.geometry.KDTreeSearchParamHybrid(radius=0.01, max_nn=30)
    )

    _, eig_src = np.linalg.eigh(cov_src)
    _, eig_tgt = np.linalg.eigh(cov_tgt)

    # 2. 粗略对齐:让 source 的主轴与 target 对齐
    R0 = eig_tgt @ eig_src.T
    t0 = c_tgt - R0 @ c_src
    T0 = np.eye(4)
    T0[:3, :3] = R0
    T0[:3, 3] = t0

    # 3. 变换
    T_total = T0
    src_coarse = copy.deepcopy(source).transform(T0)

    logging.info("PCA配准矩阵:\n%s", T_total)
    return src_coarse


# ------------------------------------------------------------------------------
# 主程序
# ------------------------------------------------------------------------------
def main() -> None:
    np.random.seed(42)

    # 1. 读入源点云
    source = o3d.io.read_point_cloud("E:/CSDN/规则点云/bunny.pcd")

    # 2. 生成目标点云:平移 + 高斯噪声
    target = copy.deepcopy(source)
    t_noise = np.array([0.10, 0.15, 0.20])
    target.translate(t_noise, relative=True)

    points = np.asarray(target.points)
    noise = np.random.normal(0, 0.001, points.shape)
    points += noise
    target.points = o3d.utility.Vector3dVector(points)

    # 3. 可视化原始
    source.paint_uniform_color([1, 0, 0])
    target.paint_uniform_color([0, 1, 0])
    o3d.visualization.draw_geometries(
        [source, target],
        window_name="原始点云",
        width=1024,
        height=768,
    )

    # 4. 配准
    registered = pca_registration(source, target)

    # 5. 可视化结果
    o3d.visualization.draw_geometries(
        [registered, target],
        window_name="PCA配准点云",
        width=1024,
        height=768,
    )


if __name__ == "__main__":
    main()

二、PCA配准结果

        可以看出,前两次的兔砸数据也能很好的配准。但是此次有一点进行了改进,就是对原始点云和目标点云进行了KD-tree处理,所以运行起来会很快。总的来说此次配准还不错,不过要说几种配准方法的优劣,还需进一步结合数据分析。

就酱,下次见^-^


网站公告

今日签到

点亮在社区的每一天
去签到