mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 06:03:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #17735 from pemmanuelviel:pev-fix-trees-descent

Browse Source 
* Fix trees parsing behavior in hierarchical_clustering_index:
Before, when maxCheck was reached in the first descent of a tree, time was still wasted parsing
the next trees till their best leaf, just to skip the points stored there.
Now we can choose either to keep this behavior, and so we skip parsing other trees after reaching
maxCheck, or we choose to do one descent in each tree, even if in one tree we reach maxCheck.

* Apply the same change to kdtree.
As each leaf contains only 1 point (unlike hierarchical_clustering), difference is visible if trees > maxCheck

* Add the new explore_all_trees parameters to miniflann

* Adapt the FlannBasedMatcher read_write test to the additional search parameter

* Adapt java tests to the additional parameter in SearchParams

* Fix the ABI dumps failure on SearchParams interface change

* Support of ctor calling another ctor of the class is only fully supported from C+11
This commit is contained in:
pemmanuelviel 2020-08-03 20:00:59 +02:00 committed by GitHub
parent 1192734131
commit e6ec42d462
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 76 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/features2d/misc/java/test/FlannBasedDescriptorMatcherTest.java
View File

@ -43,6 +43,10 @@ public class FlannBasedDescriptorMatcherTest extends OpenCVTestCase {
+ " <type>5</type>\n"
+ " <value>0.</value></_>\n"
+ " <_>\n"
+ " <name>explore_all_trees</name>\n"
+ " <type>8</type>\n"
+ " <value>0</value></_>\n"
+ " <_>\n"
+ " <name>sorted</name>\n"
+ " <type>8</type>\n" // FLANN_INDEX_TYPE_BOOL
+ " <value>1</value></_></searchParams>\n"
@ -68,6 +72,10 @@ public class FlannBasedDescriptorMatcherTest extends OpenCVTestCase {
+ " type: 5\n"
+ " value: 0.\n"
+ " -\n"
+ " name: explore_all_trees\n"
+ " type: 8\n"
+ " value: 0\n"
+ " -\n"
+ " name: sorted\n"
+ " type: 8\n" // FLANN_INDEX_TYPE_BOOL
+ " value: 1\n";
@ -92,6 +100,10 @@ public class FlannBasedDescriptorMatcherTest extends OpenCVTestCase {
+ " type: 5\n"
+ " value: 4.\n"// this line is changed!
+ " -\n"
+ " name: explore_all_trees\n"
+ " type: 8\n"
+ " value: 1\n"// this line is changed!
+ " -\n"
+ " name: sorted\n"
+ " type: 8\n" // FLANN_INDEX_TYPE_BOOL
+ " value: 1\n";

4
modules/features2d/test/test_matchers_algorithmic.cpp
View File

@ -582,6 +582,10 @@ TEST( Features2d_FlannBasedMatcher, read_write )
" type: 5\n"
" value: 4.\n"// this line is changed!
" -\n"
" name: explore_all_trees\n"
" type: 8\n"
" value: 0\n"
" -\n"
" name: sorted\n"
" type: 8\n" // FLANN_INDEX_TYPE_BOOL
" value: 1\n";

13
modules/flann/include/opencv2/flann/hierarchical_clustering_index.h
View File

@ -548,6 +548,7 @@ public:
{
const int maxChecks = get_param(searchParams,"checks",32);
const bool explore_all_trees = get_param(searchParams,"explore_all_trees",false);
// Priority queue storing intermediate branches in the best-bin-first search
Heap<BranchSt>* heap = new Heap<BranchSt>((int)size_);
@ -555,15 +556,15 @@ public:
std::vector<bool> checked(size_,false);
int checks = 0;
for (int i=0; i<trees_; ++i) {
findNN(root[i], result, vec, checks, maxChecks, heap, checked);
if ((checks >= maxChecks) && result.full())
findNN(root[i], result, vec, checks, maxChecks, heap, checked, explore_all_trees);
if (!explore_all_trees && (checks >= maxChecks) && result.full())
break;
}
BranchSt branch;
while (heap->popMin(branch) && (checks<maxChecks || !result.full())) {
NodePtr node = branch.node;
findNN(node, result, vec, checks, maxChecks, heap, checked);
findNN(node, result, vec, checks, maxChecks, heap, checked, false);
}
delete heap;
@ -746,10 +747,10 @@ private:
void findNN(NodePtr node, ResultSet<DistanceType>& result, const ElementType* vec, int& checks, int maxChecks,
Heap<BranchSt>* heap, std::vector<bool>& checked)
Heap<BranchSt>* heap, std::vector<bool>& checked, bool explore_all_trees = false)
{
if (node->childs==NULL) {
if ((checks>=maxChecks) && result.full()) {
if (!explore_all_trees && (checks>=maxChecks) && result.full()) {
return;
}
for (int i=0; i<node->size; ++i) {
@ -778,7 +779,7 @@ private:
}
}
delete[] domain_distances;
findNN(node->childs[best_index],result,vec, checks, maxChecks, heap, checked);
findNN(node->childs[best_index],result,vec, checks, maxChecks, heap, checked, explore_all_trees);
}
}

25
modules/flann/include/opencv2/flann/kdtree_index.h
View File

@ -204,14 +204,15 @@ public:
*/
void findNeighbors(ResultSet<DistanceType>& result, const ElementType* vec, const SearchParams& searchParams) CV_OVERRIDE
{
int maxChecks = get_param(searchParams,"checks", 32);
float epsError = 1+get_param(searchParams,"eps",0.0f);
const int maxChecks = get_param(searchParams,"checks", 32);
const float epsError = 1+get_param(searchParams,"eps",0.0f);
const bool explore_all_trees = get_param(searchParams,"explore_all_trees",false);
if (maxChecks==FLANN_CHECKS_UNLIMITED) {
getExactNeighbors(result, vec, epsError);
}
else {
getNeighbors(result, vec, maxChecks, epsError);
getNeighbors(result, vec, maxChecks, epsError, explore_all_trees);
}
}
@ -440,7 +441,8 @@ private:
* because the tree traversal is abandoned after a given number of descends in
* the tree.
*/
void getNeighbors(ResultSet<DistanceType>& result, const ElementType* vec, int maxCheck, float epsError)
void getNeighbors(ResultSet<DistanceType>& result, const ElementType* vec,
int maxCheck, float epsError, bool explore_all_trees = false)
{
int i;
BranchSt branch;
@ -451,12 +453,16 @@ private:
/* Search once through each tree down to root. */
for (i = 0; i < trees_; ++i) {
searchLevel(result, vec, tree_roots_[i], 0, checkCount, maxCheck, epsError, heap, checked);
searchLevel(result, vec, tree_roots_[i], 0, checkCount, maxCheck,
epsError, heap, checked, explore_all_trees);
if (!explore_all_trees && (checkCount >= maxCheck) && result.full())
break;
}
/* Keep searching other branches from heap until finished. */
while ( heap->popMin(branch) && (checkCount < maxCheck || !result.full() )) {
searchLevel(result, vec, branch.node, branch.mindist, checkCount, maxCheck, epsError, heap, checked);
searchLevel(result, vec, branch.node, branch.mindist, checkCount, maxCheck,
epsError, heap, checked, false);
}
delete heap;
@ -471,7 +477,7 @@ private:
* at least "mindistsq".
*/
void searchLevel(ResultSet<DistanceType>& result_set, const ElementType* vec, NodePtr node, DistanceType mindist, int& checkCount, int maxCheck,
float epsError, Heap<BranchSt>* heap, DynamicBitset& checked)
float epsError, Heap<BranchSt>* heap, DynamicBitset& checked, bool explore_all_trees = false)
{
if (result_set.worstDist()<mindist) {
// printf("Ignoring branch, too far\n");
@ -485,7 +491,10 @@ private:
current checkID.
*/
int index = node->divfeat;
if ( checked.test(index) || ((checkCount>=maxCheck)&& result_set.full()) ) return;
if ( checked.test(index) ||
(!explore_all_trees && (checkCount>=maxCheck) && result_set.full()) ) {
return;
}
checked.set(index);
checkCount++;

1
modules/flann/include/opencv2/flann/miniflann.hpp
View File

@ -143,6 +143,7 @@ struct CV_EXPORTS SavedIndexParams : public IndexParams
struct CV_EXPORTS SearchParams : public IndexParams
{
SearchParams( int checks, float eps, bool sorted, bool explore_all_trees );
SearchParams( int checks = 32, float eps = 0, bool sorted = true );
};

14
modules/flann/include/opencv2/flann/params.h
View File

@ -46,6 +46,16 @@ typedef std::map<cv::String, any> IndexParams;
struct SearchParams : public IndexParams
{
SearchParams(int checks = 32, float eps = 0, bool sorted = true )
{
init(checks, eps, sorted, false);
}
SearchParams(int checks, float eps, bool sorted, bool explore_all_trees )
{
init(checks, eps, sorted, explore_all_trees);
}
void init(int checks = 32, float eps = 0, bool sorted = true, bool explore_all_trees = false )
{
// how many leafs to visit when searching for neighbours (-1 for unlimited)
(*this)["checks"] = checks;
@ -53,6 +63,10 @@ struct SearchParams : public IndexParams
(*this)["eps"] = eps;
// only for radius search, require neighbours sorted by distance (default: true)
(*this)["sorted"] = sorted;
// if false, search stops at the tree reaching the number of max checks (original behavior).
// When true, we do a descent in each tree and. Like before the alternative paths
// stored in the heap are not be processed further when max checks is reached.
(*this)["explore_all_trees"] = explore_all_trees;
}
};

21
modules/flann/src/miniflann.cpp
View File

@ -294,6 +294,23 @@ SavedIndexParams::SavedIndexParams(const String& _filename)
p["filename"] = filename;
}
SearchParams::SearchParams( int checks, float eps, bool sorted, bool explore_all_trees )
{
::cvflann::IndexParams& p = get_params(*this);
// how many leafs to visit when searching for neighbours (-1 for unlimited)
p["checks"] = checks;
// search for eps-approximate neighbours (default: 0)
p["eps"] = eps;
// only for radius search, require neighbours sorted by distance (default: true)
p["sorted"] = sorted;
// if false, search stops at the tree reaching the number of max checks (original behavior).
// When true, we do a descent in each tree and. Like before the alternative paths
// stored in the heap are not be processed further when max checks is reached.
p["explore_all_trees"] = explore_all_trees;
}
SearchParams::SearchParams( int checks, float eps, bool sorted )
{
::cvflann::IndexParams& p = get_params(*this);
@ -304,6 +321,10 @@ SearchParams::SearchParams( int checks, float eps, bool sorted )
p["eps"] = eps;
// only for radius search, require neighbours sorted by distance (default: true)
p["sorted"] = sorted;
// if false, search stops at the tree reaching the number of max checks (original behavior).
// When true, we do a descent in each tree and. Like before the alternative paths
// stored in the heap are not be processed further when max checks is reached.
p["explore_all_trees"] = false;
}