CloudCompare二次开发之如何通过PCL进行点云分割?

文章目录
0.引言
1.CloudCompare界面设计配准(segment)按钮
2.欧式聚类分割(Euclidean_Seg)
3.基于区域生长的分割(Region_Seg)

0.引言

  因笔者课题涉及点云处理,需要通过PCL进行点云数据一系列处理分析,查阅现有网络资料,对常用PCL点云分割进行代码实现,本文记录分割实现过程。

1.CloudCompare界面设计配准(segment)按钮

  (1)设计.ui文件

  ①设计按钮

②编译.ui

(2)修改mainwindow.h文件

//点云分割
void doActionPCLEuclidean_Seg(); // 欧式聚类分割  
void doActionPCLRegion_Seg(); // 基于区域生长的分割

(3)修改mainwindow.cpp文件

  ①添加头文件

    #include <pcl/ModelCoefficients.h>
    #include <pcl/sample_consensus/method_types.h>//模型定义头文件  
    #include <pcl/sample_consensus/model_types.h>//随机参数估计方法头文件  
    #include <pcl/segmentation/sac_segmentation.h>//基于采样一致性分割的类的头文件  
    #include <pcl/segmentation/extract_clusters.h>  
    #include <pcl/segmentation/region_growing.h>//区域生成分割的头文件  
    #include <pcl/filters/extract_indices.h>

②添加实现代码

//欧式聚类分割
void MainWindow::doActionPCLEuclidean_Seg()  
{  
}  
//基于区域生长的分割  
void MainWindow::PCLRegion_Seg()  
{  
}

③添加信号槽函数

connect(m_UI->actionEuclidean_Seg, &amp;QAction::triggered, this, &amp;MainWindow::doActionPCLEuclidean_Seg);//欧式聚类分割
connect(m_UI->actionRegion_Seg, &amp;QAction::triggered, this, &amp;MainWindow::doActionPCLRegion_Seg);//基于区域生长的分割

 (4)生成

2.欧式聚类分割(Euclidean_Seg)

  (1)实现代码

//欧式聚类分割
void MainWindow::doActionPCLEuclidean_Seg()  
{  
    if (getSelectedEntities().size() != 1)  
    {  
        ccLog::Print(QStringLiteral("只能选择一个点云实体"));  
    return;  
    }  
    ccHObject* entity = getSelectedEntities()[0];  
    ccPointCloud* ccCloud = ccHObjectCaster::ToPointCloud(entity);  
    // ---------------------------读取数据到PCL----------------------------------  
    pcl::PointCloud<pcl::PointXYZ>::Ptr cloud(new pcl::PointCloud<pcl::PointXYZ>);  
    cloud->resize(ccCloud->size());  
    pcl::PointCloud<pcl::PointNormal>::Ptr incloud(new pcl::PointCloud <pcl::PointNormal>());  
    for (int i = 0; i < cloud->size(); ++i)  
    {  
        const CCVector3* point = ccCloud->getPoint(i);  
        cloud->points[i].x = point->x;  
        cloud->points[i].y = point->y;  
        cloud->points[i].z = point->z;  
    pcl::PointNormal pt;  
    pt.x = point->x;  
    pt.y = point->y;  
    pt.z = point->z;  
    incloud->points.push_back(pt);  
    }  
    pcl::VoxelGrid<pcl::PointXYZ> vg;  
    pcl::PointCloud<pcl::PointXYZ>::Ptr cloud_filtered(new pcl::PointCloud<pcl::PointXYZ>);  
    vg.setInputCloud(cloud);  
    vg.setLeafSize(0.05f, 0.05f, 0.05f);  
    vg.filter(*cloud_filtered);  
    pcl::SACSegmentation<pcl::PointXYZ> seg;//实例化一个分割对象  
    pcl::PointIndices::Ptr inliers(new pcl::PointIndices);//实例化一个索引  
    pcl::ModelCoefficients::Ptr coefficients(new pcl::ModelCoefficients);//实例化模型参数  
    pcl::PointCloud<pcl::PointXYZ>::Ptr cloud_plane(new pcl::PointCloud<pcl::PointXYZ>());//提取到的平面保存至cloud_plane  
    pcl::PointCloud<pcl::PointXYZ>::Ptr cloud_f(new pcl::PointCloud<pcl::PointXYZ>());//提取到的平面保存至cloud_plane  
    //pcl::PCDWriter writer;  
    seg.setOptimizeCoefficients(true);//参数优化  
    seg.setModelType(pcl::SACMODEL_PLANE);//模型类型:平面  
    seg.setMethodType(pcl::SAC_RANSAC);//参数估计方法  
    seg.setMaxIterations(100);//最大迭代次数  
    seg.setDistanceThreshold(0.02);//设置内点到模型的距离允许最大值  
    //创建一个文件夹来放点云  
    ccHObject* CloudGroup = new ccHObject(QString("SegmentGroup"));  
    int i = 0, nr_points = (int)cloud_filtered->points.size();//计数变量i,记下提取的平面的个数  
    dispToConsole("cloud_filtered->points.size()1:" + QString::number(cloud_filtered->points.size()), ccMainAppInterface::STD_CONSOLE_MESSAGE);  
    while (cloud_filtered->points.size() > 0.3 * nr_points)  
    {  
    // Segment the largest planar component from the remaining cloud  
    seg.setInputCloud(cloud_filtered);//设置要分割的点云  
    seg.segment(*inliers, *coefficients);//输出平面点的索引和参数  
    // Extract the planar inliers from the input cloud  
    pcl::ExtractIndices<pcl::PointXYZ> extract;//实例化一个提取对象  
    extract.setInputCloud(cloud_filtered);//设置要提取的点云  
    extract.setIndices(inliers);//根据分割得到的平面的索引提取平面  
    extract.setNegative(false);//提取内点  
      // Write the planar inliers to disk  
    extract.filter(*cloud_plane);//保存提取到的平面  
    //std::cout << "PointCloud representing the planar component: " << cloud_plane->points.size() << " data points." << std::endl;  
    //存写指针的参数  
    cloud_plane->width = cloud_plane->points.size();  
    cloud_plane->height = 1;  
    cloud_plane->resize(cloud_plane->width);  
    cloud_plane->is_dense = false;  
    ccPointCloud* newPointCloud = new ccPointCloud(QString::number(i + 1) + ".1segment");  
    for (int i = 0; i < cloud_plane->size(); ++i)  
    {  
        double x = cloud_plane->points[i].x;  
        double y = cloud_plane->points[i].y;  
        double z = cloud_plane->points[i].z;  
        newPointCloud->addPoint(CCVector3(x, y, z));  
    }  
    newPointCloud->setRGBColor(ccColor::Rgba(255, 100, 100, 255));  
    newPointCloud->setRGBColor(ccColor::Rgba(rand() % 205 + 50, rand() % 155 + 100, rand() % 105 + 150, 255));  
    newPointCloud->showColors(true);  
    CloudGroup->addChild(newPointCloud);  
    //CloudGroup->getLastChild()->setEnabled(false);  
    addToDB(newPointCloud);  
    //计数变量加1  
    i++;  
    // Remove the planar inliers, extract the rest  
    extract.setNegative(true);//提取外点(除第一个平面之外的点)  
    extract.filter(*cloud_f);//保存除平面之外的剩余点  
    cloud_filtered = cloud_f;//将剩余点作为下一次分割、提取的平面的输入点云  
    }  
    // Creating the KdTree object for the search method of the extraction  
    pcl::search::KdTree<pcl::PointXYZ>::Ptr tree(new pcl::search::KdTree<pcl::PointXYZ>);  
    tree->setInputCloud(cloud_filtered);//将无法提取平面的点云作为cloud_filtered  
    std::vector<pcl::PointIndices> cluster_indices;//保存每一种聚类,每一种聚类下还有具体的聚类的点  
    pcl::EuclideanClusterExtraction<pcl::PointXYZ> ec;//实例化一个欧式聚类提取对象  
    ec.setClusterTolerance(0.02); // 近邻搜索的搜索半径为2cm,重要参数  
    ec.setMinClusterSize(100);//设置一个聚类需要的最少点数目为100  
    ec.setMaxClusterSize(25000);//一个聚类最大点数目为25000  
    ec.setSearchMethod(tree);//设置点云的搜索机制  
    ec.setInputCloud(cloud_filtered);//设置输入点云  
    ec.extract(cluster_indices);//将聚类结果保存至cluster_indices中  
    //迭代访问点云索引cluster_indices,直到分割出所有聚类,一个循环提取出一类点云  
    int j = 0;  
    for (std::vector<pcl::PointIndices>::const_iterator it = cluster_indices.begin(); it != cluster_indices.end(); ++it)  
    {  
        pcl::PointCloud<pcl::PointXYZ>::Ptr cloud_cluster(new pcl::PointCloud<pcl::PointXYZ>);  
        //创建新的点云数据集cloud_cluster,直到分割出所有聚类  
        for (std::vector<int>::const_iterator pit = it->indices.begin(); pit != it->indices.end(); pit++)  
            cloud_cluster->points.push_back(cloud_filtered->points[*pit]); //*  

        cloud_cluster->width = cloud_cluster->points.size();  
        cloud_cluster->height = 1;  
        cloud_cluster->is_dense = true;  
        ccPointCloud* newPointCloud = new ccPointCloud(QString::number(i + 1) + ".2segment");  
        for (int i = 0; i < cloud_cluster->size(); ++i)  
        {  
            double x = cloud_cluster->points[i].x;  
            double y = cloud_cluster->points[i].y;  
            double z = cloud_cluster->points[i].z;  
            newPointCloud->addPoint(CCVector3(x, y, z));  
        }  
        newPointCloud->setRGBColor(ccColor::Rgba(255, 255, 255, 255));  
        newPointCloud->showColors(true);  
    CloudGroup->addChild(newPointCloud);  
    CloudGroup->getLastChild()->setEnabled(false);  
    addToDB(newPointCloud);  
    j++;  
    }  
    m_ccRoot->addElement(CloudGroup);  
}

(2)分割结果

 ①分割前

②分割后

3.基于区域生长的分割(Region_Seg)

  (1)实现代码

//基于区域生长的分割
void MainWindow::doActionPCLRegion_Seg()  
{  
    if (getSelectedEntities().size() != 1)  
    {  
        ccLog::Print(QStringLiteral("只能选择一个点云实体"));  
    return;  
    }  
    ccHObject* entity = getSelectedEntities()[0];  
    ccPointCloud* ccCloud = ccHObjectCaster::ToPointCloud(entity);  
    // ---------------------------读取数据到PCL----------------------------------  
    pcl::PointCloud<pcl::PointXYZ>::Ptr cloud(new pcl::PointCloud<pcl::PointXYZ>);  
    cloud->resize(ccCloud->size());  
    pcl::PointCloud<pcl::PointNormal>::Ptr incloud(new pcl::PointCloud <pcl::PointNormal>());  
    for (int i = 0; i < cloud->size(); ++i)  
    {  
        const CCVector3* point = ccCloud->getPoint(i);  
        cloud->points[i].x = point->x;  
        cloud->points[i].y = point->y;  
        cloud->points[i].z = point->z;  
    pcl::PointNormal pt;  
    pt.x = point->x;  
    pt.y = point->y;  
    pt.z = point->z;  
    incloud->points.push_back(pt);  
    }  
    int KN_normal = 50; //设置默认输入参数  
    bool Bool_Cuting = false;//设置默认输入参数  
    float far_cuting = 10, near_cuting = 0, SmoothnessThreshold = 30.0, CurvatureThreshold = 0.05;//设置默认输入参数  
    pcl::search::Search<pcl::PointXYZ>::Ptr tree = boost::shared_ptr<pcl::search::Search<pcl::PointXYZ> >(new pcl::search::KdTree<pcl::PointXYZ>);//创建一个指向kd树搜索对象的共享指针  
    pcl::PointCloud <pcl::Normal>::Ptr normals(new pcl::PointCloud <pcl::Normal>);  
    pcl::NormalEstimation<pcl::PointXYZ, pcl::Normal> normal_estimator;//创建法线估计对象  
    normal_estimator.setSearchMethod(tree);//设置搜索方法  
    normal_estimator.setInputCloud(cloud);//设置法线估计对象输入点集  
    normal_estimator.setKSearch(KN_normal);// 设置用于法向量估计的k近邻数目  
    normal_estimator.compute(*normals);//计算并输出法向量  
    // 区域生长算法的5个参数  
    pcl::RegionGrowing<pcl::PointXYZ, pcl::Normal> reg;//创建区域生长分割对象  
    reg.setMinClusterSize(50);//设置一个聚类需要的最小点数  
    reg.setMaxClusterSize(1000000);//设置一个聚类需要的最大点数  
    reg.setSearchMethod(tree);//设置搜索方法  
    reg.setNumberOfNeighbours(30);//设置搜索的临近点数目  
    reg.setInputCloud(cloud);//设置输入点云  
    reg.setInputNormals(normals);//设置输入点云的法向量  
    reg.setSmoothnessThreshold(SmoothnessThreshold / 180.0 * M_PI);//设置平滑阈值  
    reg.setCurvatureThreshold(CurvatureThreshold);//设置曲率阈值  
    std::vector <pcl::PointIndices> clusters;//保存每一种聚类,每一种聚类下面还有具体的点  
    reg.extract(clusters);//获取聚类的结果,分割结果保存在点云索引的向量中。  
    //创建一个文件夹来放点云  
    ccHObject* CloudGroup = new ccHObject(QString("SegmentGroup"));  
    for (size_t i = 0; i < clusters.size(); i++)  
    {  
        pcl::PointCloud<pcl::PointXYZ>::Ptr cloud_seg1(new pcl::PointCloud<pcl::PointXYZ>);  
        for (std::vector<int>::const_iterator pit = clusters[i].indices.begin(); pit != clusters[i].indices.end(); pit++)//创建一个迭代器pit以访问第一个聚类的每一个点  
        {  
            cloud_seg1->points.push_back(cloud->points[*pit]);//迭代器pit类似于一个指针,将第一个聚类分割中的每一个点进行强制类型转换,并放置在points中  
        }  
        cloud_seg1->width = cloud_seg1->points.size();  
        cloud_seg1->height = 1;  
        cloud_seg1->is_dense = false;  
    ccPointCloud* newPointCloud = new ccPointCloud(QString::number(i + 1) + ".Region_Seg");  
    for (int i = 0; i < cloud_seg1->size(); ++i)  
    {  
        double x = cloud_seg1->points[i].x;  
        double y = cloud_seg1->points[i].y;  
        double z = cloud_seg1->points[i].z;  
        newPointCloud->addPoint(CCVector3(x, y, z));  
    }  
    //newPointCloud->setRGBColor(ccColor::Rgba(100, 255, 100, 255));  
    newPointCloud->setRGBColor(ccColor::Rgba(rand() % 105 + 150, rand() % 155 + 100, rand() % 205 + 50, 255));  
    newPointCloud->showColors(true);  
    CloudGroup->addChild(newPointCloud);  
    //CloudGroup->getLastChild()->setEnabled(false);  
    addToDB(newPointCloud);  
    }  
    m_ccRoot->addElement(CloudGroup);  
}

(2)分割结果

  ①分割前

②分割后

参考资料:
[1] 来吧!我在未来等你!. CloudCompare二次开发之如何配置PCL点云库?; 2023-05-15 [accessed 2023-05-17].
[2] Tech沉思录. PCL 【点云分割】; 2019-08-17 [accessed 2023-05-17].
[3] 步子小不扯淡. PCL: Segmentation模块之SACSegmentation点云分割; 2014-09-07 [accessed 2023-05-17].
[4] SOC罗三炮. PCL教程-点云分割之区域生长分割算法; 2023-01-08 [accessed 2023-05-17].
[5] 悠缘之空. PCL函数库摘要——点云分割; 2021-11-07 [accessed 2023-05-17].