first implementation KNearest wrapper on KDTree

This commit is contained in:
Dmitriy Anisimov
2014-08-23 18:41:32 +04:00
parent 37b1a7560c
commit 9ddb23e025
3 changed files with 173 additions and 7 deletions

View File

@@ -230,10 +230,11 @@ public:
class CV_EXPORTS_W_MAP Params
{
public:
Params(int defaultK=10, bool isclassifier=true);
Params(int defaultK=10, bool isclassifier_=true, int Emax_=INT_MAX);
CV_PROP_RW int defaultK;
CV_PROP_RW bool isclassifier;
CV_PROP_RW int Emax; // for implementation with KDTree
};
virtual void setParams(const Params& p) = 0;
virtual Params getParams() const = 0;
@@ -241,7 +242,10 @@ public:
OutputArray results,
OutputArray neighborResponses=noArray(),
OutputArray dist=noArray() ) const = 0;
static Ptr<KNearest> create(const Params& params=Params());
enum { DEFAULT=1, KDTREE=2 };
static Ptr<KNearest> create(const Params& params=Params(), int type=DEFAULT);
};
/****************************************************************************************\