CODEWITHSUNDEEP
c++filterpoint-cloud-library
DXnima
DXnima

Reputation: 21

How to do color filtering with PCL

I am learning to use pcl.I want to filter out point clouds whose color is red(rgb 255,0,0),But not work.what should I do? The PCL version I am using is 1.12.1.

#include <pcl/point_types.h>
#include <pcl/filters/conditional_removal.h>
int main()
{
    pcl::PointCloud<pcl::PointXYZRGB>::Ptr cloud(new pcl::PointCloud<pcl::PointXYZRGB>);
    pcl::PointCloud<pcl::PointXYZRGB>::Ptr cloud_filtered(new 
    pcl::PointCloud<pcl::PointXYZRGB>);

    cloud->width = 5;
    cloud->height = 1;
    cloud->points.resize((cloud->width) * (cloud->height));
    //creat point cloud
    for (size_t i = 0; i < cloud->points.size(); ++i)
    {
       cloud->points[i].x = 1024 * rand() / (RAND_MAX + 1.0f) / 1000;
       cloud->points[i].y = 1024 * rand() / (RAND_MAX + 1.0f) / 1000;
       cloud->points[i].z = 1024 * rand() / (RAND_MAX + 1.0f) / 1000;
       cloud->points[i].r = 110;
       cloud->points[i].g = 110;
       cloud->points[i].b = 110;
    }
   //set rbg 255,0,0
    cloud->points[2].r = 255;cloud->points[2].g = 0;cloud->points[2].b = 0;
    
    pcl::ConditionAnd<pcl::PointXYZRGB>::Ptr range_cond(new pcl::ConditionAnd<pcl::PointXYZRGB>());
    range_cond->addComparison(pcl::FieldComparison<pcl::PointXYZRGB>::ConstPtr(new pcl::FieldComparison<pcl::PointXYZRGB>("r", pcl::ComparisonOps::EQ, 255)));
    range_cond->addComparison(pcl::FieldComparison<pcl::PointXYZRGB>::ConstPtr(new pcl::FieldComparison<pcl::PointXYZRGB>("g", pcl::ComparisonOps::EQ, 0)));
    range_cond->addComparison(pcl::FieldComparison<pcl::PointXYZRGB>::ConstPtr(new pcl::FieldComparison<pcl::PointXYZRGB>("b", pcl::ComparisonOps::EQ, 0)));
   
   pcl::ConditionalRemoval<pcl::PointXYZRGB> condrem;
   condrem.setCondition(range_cond);
   condrem.setInputCloud(cloud);
   condrem.setKeepOrganized(true);
   condrem.filter(*cloud_filtered);
   
   std::cerr << "Cloud before filtering: " << std::endl;
   for (size_t i = 0; i < cloud->points.size(); ++i)
       std::cerr << "    " << cloud->points[i].x << " "
       << cloud->points[i].y << " "
       << cloud->points[i].z << " "
       << (int)cloud->points[i].r << " "
       << (int)cloud->points[i].g << " "
       << (int)cloud->points[i].b << std::endl;

   std::cerr << "Cloud after filtering: " << std::endl;
   for (size_t i = 0; i < cloud_filtered->points.size(); ++i)
      std::cerr << "    " << cloud_filtered->points[i].x << " "
      << cloud_filtered->points[i].y << " "
      << cloud_filtered->points[i].z << " "
      << (int)cloud_filtered->points[i].r << " "
      << (int)cloud_filtered->points[i].g << " "
      << (int)cloud_filtered->points[i].b << std::endl;
   return (0);

}

Cloud after filtering is None. result： enter image description here

Upvotes: 1

Views: 619

Answers (2)

DXnima
DXnima

Reputation: 21

My modified code is here:

//creat point cloud
for (size_t i = 0; i < cloud->points.size(); ++i)
{
   cloud->points[i].x = 1024 * rand() / (RAND_MAX + 1.0f) / 1000;
   cloud->points[i].y = 1024 * rand() / (RAND_MAX + 1.0f) / 1000;
   cloud->points[i].z = 1024 * rand() / (RAND_MAX + 1.0f) / 1000;
   cstd::uint32_t rgb = ((std::uint32_t)110 << 16 | (std::uint32_t)110 << 8 | (std::uint32_t)110);
   cloud->points[i].rgb = *reinterpret_cast<float*>(&rgb);
}
//set red rbg 255,0,0
std::uint32_t red_rgb = ((std::uint32_t)255 << 16 | (std::uint32_t)0 << 8 | (std::uint32_t)0);
cloud->points[2].rgb = *reinterpret_cast<float*>(&red_rgb);
std::cerr << " FieldsList:" << pcl::getFieldsList(*cloud) <<std::endl;

pcl::ConditionAnd<pcl::PointXYZRGB>::Ptr range_cond(new pcl::ConditionAnd<pcl::PointXYZRGB>());
std::uint32_t rgb_filter = ((std::uint32_t)255 << 16 | (std::uint32_t)0 << 8 | (std::uint32_t)0);

//Add a comparison operator less than (255,0,0) on the rgb field
range_cond->addComparison(pcl::FieldComparison<pcl::PointXYZRGB>::ConstPtr(new pcl::FieldComparison<pcl::PointXYZRGB>("rgb", pcl::ComparisonOps::LT, *reinterpret_cast<float*>(&rgb_filter))));

Final result： enter image description here

Upvotes: 1

IBitMyBytes
IBitMyBytes

Reputation: 706

The warning message gives you a hint: "field not found!" (three times). PointXYZRGB does not have r, g, and b fields. You can use getFields() to find out which fields a point type has. PointXYZRGB has a combined field rgb that you can use for filtering. However, you might want to consider using PointXYZRGBA instead (with field rgba) because the other point type uses a float for storage (historic reasons) and I am not sure how well that works with filtering.

Upvotes: 1

Related Questions