Open3D (C++ API)
0.14.1
|
Faiss for nearest neighbor search. More...
#include <FaissIndex.h>
Public Member Functions | |
FaissIndex () | |
Default Constructor. More... | |
FaissIndex (const Tensor &dataset_points) | |
Parameterized Constructor. More... | |
~FaissIndex () | |
FaissIndex (const FaissIndex &)=delete | |
FaissIndex & | operator= (const FaissIndex &)=delete |
bool | SetTensorData (const Tensor &dataset_points) override |
bool | SetTensorData (const Tensor &dataset_points, double radius) override |
std::pair< Tensor, Tensor > | SearchKnn (const Tensor &query_points, int knn) const override |
std::tuple< Tensor, Tensor, Tensor > | SearchRadius (const Tensor &query_points, const Tensor &radii, bool sort) const override |
std::tuple< Tensor, Tensor, Tensor > | SearchRadius (const Tensor &query_points, double radius, bool sort) const override |
std::tuple< Tensor, Tensor, Tensor > | SearchHybrid (const Tensor &query_points, double radius, int max_knn) const override |
![]() | |
NNSIndex () | |
Default Constructor. More... | |
virtual | ~NNSIndex () |
NNSIndex (const NNSIndex &)=delete | |
NNSIndex & | operator= (const NNSIndex &)=delete |
int | GetDimension () const |
size_t | GetDatasetSize () const |
Dtype | GetDtype () const |
Device | GetDevice () const |
Protected Attributes | |
std::unique_ptr< faiss::Index > | index |
![]() | |
Tensor | dataset_points_ |
Faiss for nearest neighbor search.
open3d::core::nns::FaissIndex::FaissIndex | ( | ) |
Default Constructor.
open3d::core::nns::FaissIndex::FaissIndex | ( | const Tensor & | dataset_points | ) |
Parameterized Constructor.
dataset_points | Provides tensor from which Faiss Index is constructed. |
open3d::core::nns::FaissIndex::~FaissIndex | ( | ) |
|
delete |
|
delete |
|
overridevirtual |
Perform hybrid search.
query_points | Query points. Must be Float32, 2D, with shape {n, d}. |
radius | Radius. |
max_knn | Maximum number of neighbor to search per query point. |
Implements open3d::core::nns::NNSIndex.
|
overridevirtual |
Perform K nearest neighbor search.
query_points | Query points. Must be Float32, 2D, with shape {n, d}. |
knn | Number of nearest neighbor to search. |
Implements open3d::core::nns::NNSIndex.
|
inlineoverridevirtual |
Perform radius search with multiple radii.
query_points | Query points. Must be Float32, 2D, with shape {n, d}. |
radii | list of radius. Must be 1D, with shape {n, }. |
Implements open3d::core::nns::NNSIndex.
|
inlineoverridevirtual |
Perform radius search.
query_points | Query points. Must be Float32, 2D, with shape {n, d}. |
radius | Radius. |
Implements open3d::core::nns::NNSIndex.
|
overridevirtual |
Set the data for the nearest neighbor search.
dataset_points | Dataset points for KDTree construction. Must be 2D, with shape {n, d}. |
Implements open3d::core::nns::NNSIndex.
|
inlineoverridevirtual |
Set the data for the nearest neighbor search.
dataset_points | Dataset points for KDTree construction. Must be 2D, with shape {n, d}. |
Implements open3d::core::nns::NNSIndex.
|
protected |