PCL点云数据在Y轴上拟合直线

#include #include <pcl\io\ply_io.h> #include <pcl\point_types.h> #include <pcl\ModelCoefficients.h> #include <pcl\filters\extract_indices.h> #include <pcl\filters\voxel_grid.h> #include <pcl\features\normal_3d.h> #include <pcl\sample_consensus\method_types.h> #include <pcl\sample_consensus\model_types.h> #include <pcl\segmentation\sac_segmentation.h> #include <pcl\visualization\pcl_visualizer.h>

int main() { // 读取点云数据 pcl::PointCloudpcl::PointXYZ::Ptr cloud(new pcl::PointCloudpcl::PointXYZ); pcl::PLYReader reader; reader.read("D:\DIANYUNWENJIANJIA\newkruskal2_ply.ply", *cloud);

// 创建霍夫变换对象
pcl::ModelCoefficients::Ptr coefficients(new pcl::ModelCoefficients);
pcl::PointIndices::Ptr inliers(new pcl::PointIndices);
pcl::SACSegmentation<pcl::PointXYZ> seg;
seg.setOptimizeCoefficients(true);
seg.setModelType(pcl::SACMODEL_LINE);
seg.setMethodType(pcl::SAC_RANSAC);
seg.setMaxIterations(1000);
seg.setDistanceThreshold(0.04);

// 设置拟合直线的方向为y轴
Eigen::Vector3f axis(0.0, 1.0, 0.0);
seg.setAxis(axis);

// 执行霍夫变换
seg.setInputCloud(cloud);
seg.segment(*inliers, *coefficients);

// 提取直线内点
pcl::ExtractIndices<pcl::PointXYZ> extract;
extract.setInputCloud(cloud);
extract.setIndices(inliers);
extract.setNegative(false);
pcl::PointCloud<pcl::PointXYZ>::Ptr line(new pcl::PointCloud<pcl::PointXYZ>);
extract.filter(*line);

// 可视化结果
pcl::visualization::PCLVisualizer viewer("3D Viewer");
viewer.setBackgroundColor(0, 0, 0);
viewer.addPointCloud<pcl::PointXYZ>(cloud, "cloud");
viewer.addPointCloud<pcl::PointXYZ>(line, "line");
viewer.setPointCloudRenderingProperties(pcl::visualization::PCL_VISUALIZER_POINT_SIZE, 1, "cloud");
viewer.setPointCloudRenderingProperties(pcl::visualization::PCL_VISUALIZER_COLOR, 1, 0, 0, "line");
viewer.addCoordinateSystem(1.0);
viewer.initCameraParameters();
viewer.spin();

// 保存结果
pcl::PLYWriter writer;
writer.write("D:\\DIANYUNWENJIANJIA\\拟合中心线_ply.ply", *line);

return 0;

}