added BFM perf. test

This commit is contained in:
Alexey Spizhevoy
2011-01-26 15:28:42 +00:00
parent ba32b447ee
commit feff022422
2 changed files with 76 additions and 2 deletions

View File

@@ -273,4 +273,78 @@ TEST(SURF)
d_surf(d_src1, gpu::GpuMat(), d_keypoints1);
d_surf(d_src2, gpu::GpuMat(), d_keypoints2);
GPU_OFF;
}
TEST(BruteForceMatcher)
{
RNG rng(0);
// Init CPU matcher
int desc_len = 128;
int num_trains = rng.uniform(1, 5);
BruteForceMatcher< L2<float> > matcher;
Mat query;
gen(query, rng.uniform(100, 300), desc_len, CV_32F, 0, 10);
vector<Mat> trains(num_trains);
for (int i = 0; i < num_trains; ++i)
{
Mat train;
gen(train, rng.uniform(100, 300), desc_len, CV_32F, 0, 10);
trains[i] = train;
}
matcher.add(trains);
// Init GPU matcher
gpu::BruteForceMatcher_GPU< L2<float> > d_matcher;
gpu::GpuMat d_query(query);
vector<gpu::GpuMat> d_trains(num_trains);
for (int i = 0; i < num_trains; ++i)
{
d_trains[i] = trains[i];
}
d_matcher.add(d_trains);
// Output
vector< vector<DMatch> > matches(1);
vector< vector<DMatch> > d_matches(1);
SUBTEST << "match";
CPU_ON;
matcher.match(query, matches[0]);
CPU_OFF;
GPU_ON;
d_matcher.match(d_query, d_matches[0]);
GPU_OFF;
SUBTEST << "knnMatch";
int knn = rng.uniform(3, 10);
CPU_ON;
matcher.knnMatch(query, matches, knn);
CPU_OFF;
GPU_ON;
d_matcher.knnMatch(d_query, d_matches, knn);
GPU_OFF;
SUBTEST << "radiusMatch";
float max_distance = rng.uniform(25.0f, 65.0f);
CPU_ON;
matcher.radiusMatch(query, matches, max_distance);
CPU_OFF;
GPU_ON;
d_matcher.radiusMatch(d_query, d_matches, max_distance);
GPU_OFF;
}