find_object::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)
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< SuperPoint >	model_
bool	nms_
torch::Tensor	prob_
float	threshold_

Detailed Description

Definition at line 49 of file SuperPoint.h.

Constructor & Destructor Documentation

SPDetector()

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

Definition at line 113 of file SuperPoint.cc.

~SPDetector()

find_object::SPDetector::~SPDetector ( )

virtual

Definition at line 136 of file SuperPoint.cc.

Member Function Documentation

compute()

cv::Mat find_object::SPDetector::compute

(

const std::vector< cv::KeyPoint > &

keypoints

)

Definition at line 195 of file SuperPoint.cc.

detect()

std::vector< cv::KeyPoint > find_object::SPDetector::detect

(

const cv::Mat &

img

)

Definition at line 140 of file SuperPoint.cc.

setMinDistance()

void find_object::SPDetector::setMinDistance ( float  minDistance )

inline

Definition at line 58 of file SuperPoint.h.

SetNMS()

void find_object::SPDetector::SetNMS ( bool  enabled )

inline

Definition at line 57 of file SuperPoint.h.

setThreshold()

void find_object::SPDetector::setThreshold ( float  threshold )

inline

Definition at line 56 of file SuperPoint.h.

Member Data Documentation

cuda_

bool find_object::SPDetector::cuda_

private

Definition at line 68 of file SuperPoint.h.

desc_

torch::Tensor find_object::SPDetector::desc_

private

Definition at line 63 of file SuperPoint.h.

detected_

bool find_object::SPDetector::detected_

private

Definition at line 70 of file SuperPoint.h.

minDistance_

int find_object::SPDetector::minDistance_

private

Definition at line 67 of file SuperPoint.h.

model_

std::shared_ptr<SuperPoint> find_object::SPDetector::model_

private

Definition at line 61 of file SuperPoint.h.

nms_

bool find_object::SPDetector::nms_

private

Definition at line 66 of file SuperPoint.h.

prob_

torch::Tensor find_object::SPDetector::prob_

private

Definition at line 62 of file SuperPoint.h.

threshold_

float find_object::SPDetector::threshold_

private

Definition at line 65 of file SuperPoint.h.

---

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

• SuperPoint.h
• SuperPoint.cc