KPConv-PyTorch/cpp_wrappers/cpp_neighbors/neighbors/neighbors.h
2020-03-31 15:42:35 -04:00

30 lines
1 KiB
C++

#include "../../cpp_utils/cloud/cloud.h"
#include "../../cpp_utils/nanoflann/nanoflann.hpp"
#include <set>
#include <cstdint>
using namespace std;
void ordered_neighbors(vector<PointXYZ>& queries,
vector<PointXYZ>& supports,
vector<int>& neighbors_indices,
float radius);
void batch_ordered_neighbors(vector<PointXYZ>& queries,
vector<PointXYZ>& supports,
vector<int>& q_batches,
vector<int>& s_batches,
vector<int>& neighbors_indices,
float radius);
void batch_nanoflann_neighbors(vector<PointXYZ>& queries,
vector<PointXYZ>& supports,
vector<int>& q_batches,
vector<int>& s_batches,
vector<int>& neighbors_indices,
float radius);