对点云添加高斯噪声并保存到本地（PCL 附C++代码）

最新推荐文章于 2024-10-05 00:00:00 发布

McQueen_LT

最新推荐文章于 2024-10-05 00:00:00 发布

阅读量1.3k

点赞数 2

分类专栏：计算机图形学文章标签： c++ boost

本文链接：https://blog.csdn.net/mcqueen_lt/article/details/110139165

版权

计算机图形学专栏收录该内容

34 篇文章 60 订阅

订阅专栏

代码

#include <stdio.h>
#include <stdlib.h>
#include <pcl/point_types.h>
#include <iostream>
#include <pcl/io/ply_io.h>
#include <pcl/io/pcd_io.h>
#include <pcl/features/normal_3d.h>
#include <pcl/visualization/cloud_viewer.h>
#include <boost/random.hpp>
using namespace std;


void AddNoise()
{
	string cleanCloud;
	pcl::PointCloud<pcl::PointXYZRGB>::Ptr cc(new pcl::PointCloud<pcl::PointXYZRGB>);
	//pcl::PointCloud<pcl::Normal>::Ptr cloudNormals(new pcl::PointCloud<pcl::Normal>);
	cout << "input filename of clean cloud: ";
	cin >> cleanCloud;
	
	pcl::PLYReader reader;
	reader.read(cleanCloud+".ply", *cc);
	//reader.read(cleanCloud, *cloudNormals);

	float sigmaGaussian;
	cout << "input the noise level need to be added: ";
	cin >> sigmaGaussian;

	boost::mt19937 rng; 
	rng.seed(static_cast<unsigned int>(time(0)));
	boost::normal_distribution<> nd(0, sigmaGaussian);
	boost::variate_generator<boost::mt19937&, boost::normal_distribution<>> var_nor(rng, nd);

	for (size_t i = 0; i < cc->points.size(); i++)
	{
		cc->points[i].x = cc->points[i].x + static_cast<float> (var_nor());
		cc->points[i].y = cc->points[i].y + static_cast<float> (var_nor());
		cc->points[i].z = cc->points[i].z + static_cast<float> (var_nor());
		cc->at(i).r = 150;
		cc->at(i).g = 150;
		cc->at(i).b = 150;
	}

	ofstream fout("NoisyFandisk-"+std::to_string(sigmaGaussian) + ".ply");
	fout << "ply" << endl << "format ascii 1.0" << endl
		<< "element vertex " + std::to_string(cc->points.size()) << endl
		<< "property float x" << endl << "property float y" << endl << "property float z" << endl
		<< "property uchar red" << endl << "property uchar green" << endl << "property uchar blue" << endl
		<< "end_header" << endl;
	for (int i = 0; i < cc->points.size(); i++)
	{
		fout << cc->points[i].x
			<< " " << cc->points[i].y
			<< " " << cc->points[i].z
			<< " " << std::to_string(130) << " " << std::to_string(130) << " " << std::to_string(130)
			<< endl;
	}
	fout.close();

	pcl::visualization::CloudViewer viewer("view");
	viewer.showCloud(cc);
	system("PAUSE");

}

int main()
{
	AddNoise();
	return 0;
}