mirror
/
small_gicp
mirror of https://github.com/koide3/small_gicp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

return processed point cloud

This commit is contained in:
k.koide 2025-04-27 21:21:54 +09:00
parent 2c5e9e6092
commit b39b96dbc8
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/python/preprocess.cpp
View File

@ -107,7 +107,7 @@ void define_preprocess(py::module& m) {
// estimate_normals
m.def(
"estimate_normals",
[](PointCloud::Ptr points, std::shared_ptr<KdTree<PointCloud>> tree, int num_neighbors, int num_threads) {
[](PointCloud::Ptr points, std::shared_ptr<KdTree<PointCloud>> tree, int num_neighbors, int num_threads) -> std::shared_ptr<PointCloud> {
if (tree == nullptr) {
tree = std::make_shared<KdTree<PointCloud>>(points, KdTreeBuilderOMP(num_threads));
}
@ -117,6 +117,8 @@ void define_preprocess(py::module& m) {
} else {
estimate_normals_omp(*points, *tree, num_neighbors, num_threads);
}
return points;
},
py::arg("points"),
py::arg("tree") = nullptr,
@ -141,7 +143,7 @@ void define_preprocess(py::module& m) {
// estimate_covariances
m.def(
"estimate_covariances",
[](PointCloud::Ptr points, std::shared_ptr<KdTree<PointCloud>> tree, int num_neighbors, int num_threads) {
[](PointCloud::Ptr points, std::shared_ptr<KdTree<PointCloud>> tree, int num_neighbors, int num_threads) -> std::shared_ptr<PointCloud> {
if (tree == nullptr) {
tree = std::make_shared<KdTree<PointCloud>>(points, KdTreeBuilderOMP(num_threads));
}
@ -151,6 +153,8 @@ void define_preprocess(py::module& m) {
} else {
estimate_covariances_omp(*points, *tree, num_neighbors, num_threads);
}
return points;
},
py::arg("points"),
py::arg("tree") = nullptr,