-
Notifications
You must be signed in to change notification settings - Fork 16
/
Copy pathsuperpoint.h
32 lines (28 loc) · 831 Bytes
/
superpoint.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#include <vector>
#include <string>
#include <memory>
#include<iostream>
#include "opencv2/stitching.hpp"
#include "opencv2/calib3d.hpp"
#include "common.h"
using namespace cv::detail;
using namespace cv;
using namespace std;
class FEATURE_MATCHER_EXPORTS SuperPoint :public Feature2D
{
protected:
std::wstring m_modelPath;
vector<float> ApplyTransform(const Mat& image, float& mean, float& std);
public:
SuperPoint(std::wstring modelPath);
virtual void detectAndCompute(InputArray image, InputArray mask,
std::vector<KeyPoint>& keypoints,
OutputArray descriptors,
bool useProvidedKeypoints = false);
virtual void detect(InputArray image,
std::vector<KeyPoint>& keypoints,
InputArray mask = noArray());
virtual void compute(InputArray image,
std::vector<KeyPoint>& keypoints,
OutputArray descriptors);
};