added padded performance tests for gpu module
This commit is contained in:
132
modules/gpu/perf/perf_features2d.cpp
Normal file
132
modules/gpu/perf/perf_features2d.cpp
Normal file
@@ -0,0 +1,132 @@
|
||||
#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);
|
||||
}
|
||||
|
Reference in New Issue
Block a user