mirror of
https://github.com/introlab/rtabmap.git
synced 2026-09-01 17:10:26 +08:00
Added min polygon cluster size option (DbViewer and GUI export)
This commit is contained in:
@@ -1555,6 +1555,45 @@ pcl::IndicesPtr normalFiltering(
|
||||
return output;
|
||||
}
|
||||
|
||||
void colorMeanFiltering(
|
||||
pcl::PointCloud<pcl::PointXYZRGB>::Ptr & cloud,
|
||||
const pcl::IndicesPtr & indices,
|
||||
const pcl::PointCloud<pcl::PointXYZRGB>::Ptr & cloudRef,
|
||||
const pcl::IndicesPtr & indicesRef,
|
||||
float radiusSearch)
|
||||
{
|
||||
UASSERT(radiusSearch>0.0f);
|
||||
pcl::search::KdTree<pcl::PointXYZRGB>::Ptr tree (new pcl::search::KdTree<pcl::PointXYZRGB>(false));
|
||||
if(indicesRef->size())
|
||||
{
|
||||
tree->setInputCloud(cloudRef, indicesRef);
|
||||
}
|
||||
else
|
||||
{
|
||||
tree->setInputCloud(cloudRef);
|
||||
}
|
||||
for(unsigned int i=0; i<indices->size(); ++i)
|
||||
{
|
||||
std::vector<int> kIndices;
|
||||
std::vector<float> kDistances;
|
||||
pcl::PointXYZRGB & pt = cloud->at(indices->at(i));
|
||||
if(tree->radiusSearch(pt, radiusSearch, kIndices, kDistances))
|
||||
{
|
||||
UASSERT(kIndices.size());
|
||||
int r=0,g=0,b=0;
|
||||
for(unsigned int j=0; j<kIndices.size(); ++j)
|
||||
{
|
||||
r+=cloudRef->at(kIndices.at(j)).r;
|
||||
g+=cloudRef->at(kIndices.at(j)).g;
|
||||
b+=cloudRef->at(kIndices.at(j)).b;
|
||||
}
|
||||
pt.r = (unsigned char)(r/kIndices.size());
|
||||
pt.g = (unsigned char)(g/kIndices.size());
|
||||
pt.b = (unsigned char)(b/kIndices.size());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
std::vector<pcl::IndicesPtr> extractClusters(
|
||||
const pcl::PointCloud<pcl::PointXYZ>::Ptr & cloud,
|
||||
float clusterTolerance,
|
||||
|
||||
@@ -100,6 +100,43 @@ void createPolygonIndexes(
|
||||
}
|
||||
}
|
||||
|
||||
std::list<std::list<int> > clusterPolygons(
|
||||
const std::vector<std::set<int> > & neighborPolygons,
|
||||
int minClusterSize)
|
||||
{
|
||||
std::set<int> polygonsChecked;
|
||||
|
||||
std::list<std::list<int> > clusters;
|
||||
|
||||
for(unsigned int i=0; i<neighborPolygons.size(); ++i)
|
||||
{
|
||||
if(polygonsChecked.find(i) == polygonsChecked.end())
|
||||
{
|
||||
std::list<int> currentCluster;
|
||||
currentCluster.push_back(i);
|
||||
polygonsChecked.insert(i);
|
||||
|
||||
for(std::list<int>::iterator iter=currentCluster.begin(); iter!=currentCluster.end(); ++iter)
|
||||
{
|
||||
// get neighbor polygons
|
||||
std::set<int> neighbors = neighborPolygons[*iter];
|
||||
for(std::set<int>::iterator jter=neighbors.begin(); jter!=neighbors.end(); ++jter)
|
||||
{
|
||||
if(polygonsChecked.insert(*jter).second)
|
||||
{
|
||||
currentCluster.push_back(*jter);
|
||||
}
|
||||
}
|
||||
}
|
||||
if(currentCluster.size() > minClusterSize)
|
||||
{
|
||||
clusters.push_back(currentCluster);
|
||||
}
|
||||
}
|
||||
}
|
||||
return clusters;
|
||||
}
|
||||
|
||||
std::vector<pcl::Vertices> organizedFastMesh(
|
||||
const pcl::PointCloud<pcl::PointXYZRGB>::Ptr & cloud,
|
||||
double angleTolerance,
|
||||
|
||||
Reference in New Issue
Block a user