Public Member Functions | Private Attributes | List of all members
rtabmap::SPDetector Class Reference

#include <SuperPoint.h>

Public Member Functions

cv::Mat compute (const std::vector< cv::KeyPoint > &keypoints)
 
std::vector< cv::KeyPoint > detect (const cv::Mat &img, const cv::Mat &mask=cv::Mat())
 
void setMinDistance (float minDistance)
 
void SetNMS (bool enabled)
 
void setThreshold (float threshold)
 
 SPDetector (const std::string &modelPath, float threshold=0.2f, bool nms=true, int minDistance=4, bool cuda=false)
 
virtual ~SPDetector ()
 

Private Attributes

bool cuda_
 
torch::Tensor desc_
 
bool detected_
 
int minDistance_
 
std::shared_ptr< SuperPointmodel_
 
bool nms_
 
torch::Tensor prob_
 
float threshold_
 

Detailed Description

Definition at line 49 of file SuperPoint.h.

Constructor & Destructor Documentation

rtabmap::SPDetector::SPDetector ( const std::string &  modelPath,
float  threshold = 0.2f,
bool  nms = true,
int  minDistance = 4,
bool  cuda = false 
)

Definition at line 116 of file SuperPoint.cc.

rtabmap::SPDetector::~SPDetector ( )
virtual

Definition at line 146 of file SuperPoint.cc.

Member Function Documentation

cv::Mat rtabmap::SPDetector::compute ( const std::vector< cv::KeyPoint > &  keypoints)

Definition at line 214 of file SuperPoint.cc.

std::vector< cv::KeyPoint > rtabmap::SPDetector::detect ( const cv::Mat &  img,
const cv::Mat &  mask = cv::Mat() 
)

Definition at line 150 of file SuperPoint.cc.

void rtabmap::SPDetector::setMinDistance ( float  minDistance)
inline

Definition at line 58 of file SuperPoint.h.

void rtabmap::SPDetector::SetNMS ( bool  enabled)
inline

Definition at line 57 of file SuperPoint.h.

void rtabmap::SPDetector::setThreshold ( float  threshold)
inline

Definition at line 56 of file SuperPoint.h.

Member Data Documentation

bool rtabmap::SPDetector::cuda_
private

Definition at line 68 of file SuperPoint.h.

torch::Tensor rtabmap::SPDetector::desc_
private

Definition at line 63 of file SuperPoint.h.

bool rtabmap::SPDetector::detected_
private

Definition at line 70 of file SuperPoint.h.

int rtabmap::SPDetector::minDistance_
private

Definition at line 67 of file SuperPoint.h.

std::shared_ptr<SuperPoint> rtabmap::SPDetector::model_
private

Definition at line 61 of file SuperPoint.h.

bool rtabmap::SPDetector::nms_
private

Definition at line 66 of file SuperPoint.h.

torch::Tensor rtabmap::SPDetector::prob_
private

Definition at line 62 of file SuperPoint.h.

float rtabmap::SPDetector::threshold_
private

Definition at line 65 of file SuperPoint.h.


The documentation for this class was generated from the following files:


rtabmap
Author(s): Mathieu Labbe
autogenerated on Mon Dec 14 2020 03:37:09