123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- #ifndef _OPENCV_MINIFLANN_HPP_
- #define _OPENCV_MINIFLANN_HPP_
- #include "opencv2/core.hpp"
- #include "opencv2/flann/defines.h"
- namespace cv
- {
- namespace flann
- {
- struct CV_EXPORTS IndexParams
- {
- IndexParams();
- ~IndexParams();
- String getString(const String& key, const String& defaultVal=String()) const;
- int getInt(const String& key, int defaultVal=-1) const;
- double getDouble(const String& key, double defaultVal=-1) const;
- void setString(const String& key, const String& value);
- void setInt(const String& key, int value);
- void setDouble(const String& key, double value);
- void setFloat(const String& key, float value);
- void setBool(const String& key, bool value);
- void setAlgorithm(int value);
- void getAll(std::vector<String>& names,
- std::vector<int>& types,
- std::vector<String>& strValues,
- std::vector<double>& numValues) const;
- void* params;
- };
- struct CV_EXPORTS KDTreeIndexParams : public IndexParams
- {
- KDTreeIndexParams(int trees=4);
- };
- struct CV_EXPORTS LinearIndexParams : public IndexParams
- {
- LinearIndexParams();
- };
- struct CV_EXPORTS CompositeIndexParams : public IndexParams
- {
- CompositeIndexParams(int trees = 4, int branching = 32, int iterations = 11,
- cvflann::flann_centers_init_t centers_init = cvflann::FLANN_CENTERS_RANDOM, float cb_index = 0.2f );
- };
- struct CV_EXPORTS AutotunedIndexParams : public IndexParams
- {
- AutotunedIndexParams(float target_precision = 0.8f, float build_weight = 0.01f,
- float memory_weight = 0, float sample_fraction = 0.1f);
- };
- struct CV_EXPORTS HierarchicalClusteringIndexParams : public IndexParams
- {
- HierarchicalClusteringIndexParams(int branching = 32,
- cvflann::flann_centers_init_t centers_init = cvflann::FLANN_CENTERS_RANDOM, int trees = 4, int leaf_size = 100 );
- };
- struct CV_EXPORTS KMeansIndexParams : public IndexParams
- {
- KMeansIndexParams(int branching = 32, int iterations = 11,
- cvflann::flann_centers_init_t centers_init = cvflann::FLANN_CENTERS_RANDOM, float cb_index = 0.2f );
- };
- struct CV_EXPORTS LshIndexParams : public IndexParams
- {
- LshIndexParams(int table_number, int key_size, int multi_probe_level);
- };
- struct CV_EXPORTS SavedIndexParams : public IndexParams
- {
- SavedIndexParams(const String& filename);
- };
- struct CV_EXPORTS SearchParams : public IndexParams
- {
- SearchParams( int checks = 32, float eps = 0, bool sorted = true );
- };
- class CV_EXPORTS_W Index
- {
- public:
- CV_WRAP Index();
- CV_WRAP Index(InputArray features, const IndexParams& params, cvflann::flann_distance_t distType=cvflann::FLANN_DIST_L2);
- virtual ~Index();
- CV_WRAP virtual void build(InputArray features, const IndexParams& params, cvflann::flann_distance_t distType=cvflann::FLANN_DIST_L2);
- CV_WRAP virtual void knnSearch(InputArray query, OutputArray indices,
- OutputArray dists, int knn, const SearchParams& params=SearchParams());
- CV_WRAP virtual int radiusSearch(InputArray query, OutputArray indices,
- OutputArray dists, double radius, int maxResults,
- const SearchParams& params=SearchParams());
- CV_WRAP virtual void save(const String& filename) const;
- CV_WRAP virtual bool load(InputArray features, const String& filename);
- CV_WRAP virtual void release();
- CV_WRAP cvflann::flann_distance_t getDistance() const;
- CV_WRAP cvflann::flann_algorithm_t getAlgorithm() const;
- protected:
- cvflann::flann_distance_t distType;
- cvflann::flann_algorithm_t algo;
- int featureType;
- void* index;
- };
- } }
- #endif
|