mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
132 lines
3.7 KiB
132 lines
3.7 KiB
#include "perf_precomp.hpp" |
|
|
|
PERF_TEST_P(DevInfo_DescSize, BruteForceMatcher_match, testing::Combine(testing::ValuesIn(devices()), |
|
testing::Values(64, 128))) |
|
{ |
|
DeviceInfo devInfo = std::tr1::get<0>(GetParam()); |
|
int desc_size = std::tr1::get<1>(GetParam()); |
|
|
|
setDevice(devInfo.deviceID()); |
|
|
|
Mat query_host(3000, desc_size, CV_32FC1); |
|
Mat train_host(3000, desc_size, CV_32FC1); |
|
|
|
declare.in(query_host, train_host, WARMUP_RNG); |
|
|
|
GpuMat query(query_host); |
|
GpuMat train(train_host); |
|
GpuMat trainIdx, distance; |
|
|
|
BruteForceMatcher_GPU< L2<float> > matcher; |
|
|
|
declare.time(0.5).iterations(100); |
|
|
|
SIMPLE_TEST_CYCLE() |
|
{ |
|
matcher.matchSingle(query, train, trainIdx, distance); |
|
} |
|
|
|
Mat trainIdx_host(trainIdx); |
|
Mat distance_host(distance); |
|
|
|
SANITY_CHECK(trainIdx_host); |
|
SANITY_CHECK(distance_host); |
|
} |
|
|
|
PERF_TEST_P(DevInfo_K_DescSize, BruteForceMatcher_knnMatch, testing::Combine(testing::ValuesIn(devices()), |
|
testing::Values(2, 3), |
|
testing::Values(64, 128))) |
|
{ |
|
DeviceInfo devInfo = std::tr1::get<0>(GetParam()); |
|
int k = std::tr1::get<1>(GetParam()); |
|
int desc_size = std::tr1::get<2>(GetParam()); |
|
|
|
setDevice(devInfo.deviceID()); |
|
|
|
Mat query_host(3000, desc_size, CV_32FC1); |
|
Mat train_host(3000, desc_size, CV_32FC1); |
|
|
|
declare.in(query_host, train_host, WARMUP_RNG); |
|
|
|
GpuMat query(query_host); |
|
GpuMat train(train_host); |
|
GpuMat trainIdx, distance, allDist; |
|
|
|
BruteForceMatcher_GPU< L2<float> > matcher; |
|
|
|
declare.time(0.5).iterations(100); |
|
|
|
SIMPLE_TEST_CYCLE() |
|
{ |
|
matcher.knnMatch(query, train, trainIdx, distance, allDist, k); |
|
} |
|
|
|
Mat trainIdx_host(trainIdx); |
|
Mat distance_host(distance); |
|
|
|
SANITY_CHECK(trainIdx_host); |
|
SANITY_CHECK(distance_host); |
|
} |
|
|
|
PERF_TEST_P(DevInfo_DescSize, BruteForceMatcher_radiusMatch, testing::Combine(testing::ValuesIn(devices(GLOBAL_ATOMICS)), |
|
testing::Values(64, 128))) |
|
{ |
|
DeviceInfo devInfo = std::tr1::get<0>(GetParam()); |
|
int desc_size = std::tr1::get<1>(GetParam()); |
|
|
|
setDevice(devInfo.deviceID()); |
|
|
|
Mat query_host = cvtest::randomMat(theRNG(), Size(desc_size, 3000), CV_32FC1, 0, 1, false); |
|
Mat train_host = cvtest::randomMat(theRNG(), Size(desc_size, 3000), CV_32FC1, 0, 1, false); |
|
|
|
GpuMat query(query_host); |
|
GpuMat train(train_host); |
|
GpuMat trainIdx, nMatches, distance; |
|
|
|
BruteForceMatcher_GPU< L2<float> > matcher; |
|
|
|
declare.time(0.5).iterations(100); |
|
|
|
SIMPLE_TEST_CYCLE() |
|
{ |
|
matcher.radiusMatch(query, train, trainIdx, nMatches, distance, 2.0); |
|
} |
|
|
|
Mat trainIdx_host(trainIdx); |
|
Mat nMatches_host(nMatches); |
|
Mat distance_host(distance); |
|
|
|
SANITY_CHECK(trainIdx_host); |
|
SANITY_CHECK(nMatches_host); |
|
SANITY_CHECK(distance_host); |
|
} |
|
|
|
PERF_TEST_P(DevInfo, SURF, testing::ValuesIn(devices())) |
|
{ |
|
DeviceInfo devInfo = GetParam(); |
|
|
|
setDevice(devInfo.deviceID()); |
|
|
|
Mat img_host = readImage("gpu/perf/aloe.jpg", CV_LOAD_IMAGE_GRAYSCALE); |
|
|
|
ASSERT_FALSE(img_host.empty()); |
|
|
|
GpuMat img(img_host); |
|
GpuMat keypoints, descriptors; |
|
|
|
SURF_GPU surf; |
|
|
|
declare.time(2.0); |
|
|
|
SIMPLE_TEST_CYCLE() |
|
{ |
|
surf(img, GpuMat(), keypoints, descriptors); |
|
} |
|
|
|
Mat keypoints_host(keypoints); |
|
Mat descriptors_host(descriptors); |
|
|
|
SANITY_CHECK(keypoints_host); |
|
SANITY_CHECK(descriptors_host); |
|
} |
|
|
|
|