opencv/modules/stitching/matchers.hpp

84 lines
2.4 KiB
C++
Raw Normal View History

2011-05-06 10:05:02 +02:00
#ifndef __OPENCV_MATCHERS_HPP__
#define __OPENCV_MATCHERS_HPP__
#include <vector>
#include <opencv2/core/core.hpp>
#include <opencv2/features2d/features2d.hpp>
struct ImageFeatures
{
2011-05-20 10:08:55 +02:00
cv::Size img_size;
cv::Mat hist;
2011-05-06 10:05:02 +02:00
std::vector<cv::KeyPoint> keypoints;
cv::Mat descriptors;
};
class FeaturesFinder
{
public:
virtual ~FeaturesFinder() {}
2011-05-20 10:08:55 +02:00
void operator ()(const cv::Mat &image, ImageFeatures &features);
2011-05-06 10:05:02 +02:00
protected:
2011-05-20 10:08:55 +02:00
virtual void find(const cv::Mat &image, ImageFeatures &features) = 0;
2011-05-06 10:05:02 +02:00
};
class SurfFeaturesFinder : public FeaturesFinder
{
public:
SurfFeaturesFinder(bool try_use_gpu = true, double hess_thresh = 300.0,
int num_octaves = 3, int num_layers = 4,
int num_octaves_descr = 4, int num_layers_descr = 2);
2011-05-06 10:05:02 +02:00
protected:
2011-05-20 10:08:55 +02:00
void find(const cv::Mat &image, ImageFeatures &features);
2011-05-06 10:05:02 +02:00
cv::Ptr<FeaturesFinder> impl_;
};
struct MatchesInfo
{
MatchesInfo();
MatchesInfo(const MatchesInfo &other);
const MatchesInfo& operator =(const MatchesInfo &other);
int src_img_idx, dst_img_idx; // Images indices (optional)
2011-05-06 10:05:02 +02:00
std::vector<cv::DMatch> matches;
std::vector<uchar> inliers_mask; // Geometrically consistent matches mask
int num_inliers; // Number of geometrically consistent matches
cv::Mat H; // Estimated homography
double confidence; // Confidence two images are from the same panorama
2011-05-06 10:05:02 +02:00
};
class FeaturesMatcher
{
public:
virtual ~FeaturesMatcher() {}
2011-05-20 10:08:55 +02:00
void operator ()(const ImageFeatures &features1, const ImageFeatures &features2, MatchesInfo& matches_info)
{ match(features1, features2, matches_info); }
void operator ()(const std::vector<ImageFeatures> &features, std::vector<MatchesInfo> &pairwise_matches);
2011-05-06 10:05:02 +02:00
protected:
2011-05-20 10:08:55 +02:00
virtual void match(const ImageFeatures &features1, const ImageFeatures &features2, MatchesInfo& matches_info) = 0;
2011-05-06 10:05:02 +02:00
};
class BestOf2NearestMatcher : public FeaturesMatcher
{
public:
BestOf2NearestMatcher(bool try_use_gpu = true, float match_conf = 0.55f, int num_matches_thresh1 = 6, int num_matches_thresh2 = 6);
2011-05-06 10:05:02 +02:00
protected:
2011-05-20 10:08:55 +02:00
void match(const ImageFeatures &features1, const ImageFeatures &features2, MatchesInfo &matches_info);
2011-05-06 10:05:02 +02:00
int num_matches_thresh1_;
int num_matches_thresh2_;
cv::Ptr<FeaturesMatcher> impl_;
};
#endif // __OPENCV_MATCHERS_HPP__