当前位置：首页 > news >正文

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

news 2025/8/9 9:49:02

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

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

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

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

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

一、PCA配准程序

from __future__ import annotationsimport copy
import logging
from typing import Tupleimport numpy as np
import open3d as o3dlogging.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.Tt0 = c_tgt - R0 @ c_srcT0 = np.eye(4)T0[:3, :3] = R0T0[:3, 3] = t0# 3. 变换T_total = T0src_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 += noisetarget.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配准结果