Commit 5421d741 authored by Andrey Pavlenko's avatar Andrey Pavlenko

making OCL tests conform to the comon style

parent 22f42a63
...@@ -11,4 +11,29 @@ static const char * impls[] = { ...@@ -11,4 +11,29 @@ static const char * impls[] = {
"plain" "plain"
}; };
CV_PERF_TEST_MAIN_WITH_IMPLS(nonfree, impls, perf::printCudaInfo()) #ifdef HAVE_OPENCL
#define DUMP_PROPERTY_XML(propertyName, propertyValue) \
do { \
std::stringstream ssName, ssValue;\
ssName << propertyName;\
ssValue << propertyValue; \
::testing::Test::RecordProperty(ssName.str(), ssValue.str()); \
} while (false)
#define DUMP_MESSAGE_STDOUT(msg) \
do { \
std::cout << msg << std::endl; \
} while (false)
#include "opencv2/ocl/private/opencl_dumpinfo.hpp"
#endif
int main(int argc, char **argv)
{
::perf::TestBase::setPerformanceStrategy(::perf::PERF_STRATEGY_SIMPLE);
#if defined(HAVE_CUDA)
CV_PERF_TEST_MAIN_INTERNALS(nonfree, impls, perf::printCudaInfo());
#else if defined(HAVE_OPENCL)
CV_PERF_TEST_MAIN_INTERNALS(nonfree, impls, dumpOpenCLDevice());
#endif
}
...@@ -9,6 +9,8 @@ ...@@ -9,6 +9,8 @@
#ifndef __OPENCV_PERF_PRECOMP_HPP__ #ifndef __OPENCV_PERF_PRECOMP_HPP__
#define __OPENCV_PERF_PRECOMP_HPP__ #define __OPENCV_PERF_PRECOMP_HPP__
#include "cvconfig.h"
#include "opencv2/ts/ts.hpp" #include "opencv2/ts/ts.hpp"
#include "opencv2/nonfree/nonfree.hpp" #include "opencv2/nonfree/nonfree.hpp"
......
...@@ -13,36 +13,34 @@ typedef perf::TestBaseWithParam<std::string> surf; ...@@ -13,36 +13,34 @@ typedef perf::TestBaseWithParam<std::string> surf;
"stitching/a3.png" "stitching/a3.png"
#ifdef HAVE_OPENCV_OCL #ifdef HAVE_OPENCV_OCL
static Ptr<Feature2D> getSURF() #define OCL_TEST_CYCLE() for( ; startTimer(), next(); cv::ocl::finish(), stopTimer())
{
ocl::PlatformsInfo p;
if(ocl::getOpenCLPlatforms(p) > 0)
return new ocl::SURF_OCL;
else
return new SURF;
}
#else
static Ptr<Feature2D> getSURF()
{
return new SURF;
}
#endif #endif
PERF_TEST_P(surf, detect, testing::Values(SURF_IMAGES)) PERF_TEST_P(surf, detect, testing::Values(SURF_IMAGES))
{ {
String filename = getDataPath(GetParam()); String filename = getDataPath(GetParam());
Mat frame = imread(filename, IMREAD_GRAYSCALE); Mat frame = imread(filename, IMREAD_GRAYSCALE);
ASSERT_FALSE(frame.empty()) << "Unable to load source image " << filename;
if (frame.empty()) declare.in(frame);
FAIL() << "Unable to load source image " << filename;
Mat mask; Mat mask;
declare.in(frame).time(90);
Ptr<Feature2D> detector = getSURF();
vector<KeyPoint> points; vector<KeyPoint> points;
Ptr<Feature2D> detector;
TEST_CYCLE() detector->operator()(frame, mask, points, noArray()); if (getSelectedImpl() == "plain")
{
detector = new SURF;
TEST_CYCLE() detector->operator()(frame, mask, points, noArray());
}
#ifdef HAVE_OPENCV_OCL
else if (getSelectedImpl() == "ocl")
{
detector = new ocl::SURF_OCL;
OCL_TEST_CYCLE() detector->operator()(frame, mask, points, noArray());
}
#endif
else CV_TEST_FAIL_NO_IMPL();
SANITY_CHECK_KEYPOINTS(points, 1e-3); SANITY_CHECK_KEYPOINTS(points, 1e-3);
} }
...@@ -51,19 +49,30 @@ PERF_TEST_P(surf, extract, testing::Values(SURF_IMAGES)) ...@@ -51,19 +49,30 @@ PERF_TEST_P(surf, extract, testing::Values(SURF_IMAGES))
{ {
String filename = getDataPath(GetParam()); String filename = getDataPath(GetParam());
Mat frame = imread(filename, IMREAD_GRAYSCALE); Mat frame = imread(filename, IMREAD_GRAYSCALE);
ASSERT_FALSE(frame.empty()) << "Unable to load source image " << filename;
if (frame.empty()) declare.in(frame);
FAIL() << "Unable to load source image " << filename;
Mat mask; Mat mask;
declare.in(frame).time(90); Ptr<Feature2D> detector;
Ptr<Feature2D> detector = getSURF();
vector<KeyPoint> points; vector<KeyPoint> points;
vector<float> descriptors; vector<float> descriptors;
detector->operator()(frame, mask, points, noArray());
TEST_CYCLE() detector->operator()(frame, mask, points, descriptors, true); if (getSelectedImpl() == "plain")
{
detector = new SURF;
detector->operator()(frame, mask, points, noArray());
TEST_CYCLE() detector->operator()(frame, mask, points, descriptors, true);
}
#ifdef HAVE_OPENCV_OCL
else if (getSelectedImpl() == "ocl")
{
detector = new ocl::SURF_OCL;
detector->operator()(frame, mask, points, noArray());
OCL_TEST_CYCLE() detector->operator()(frame, mask, points, descriptors, true);
}
#endif
else CV_TEST_FAIL_NO_IMPL();
SANITY_CHECK(descriptors, 1e-4); SANITY_CHECK(descriptors, 1e-4);
} }
...@@ -72,17 +81,29 @@ PERF_TEST_P(surf, full, testing::Values(SURF_IMAGES)) ...@@ -72,17 +81,29 @@ PERF_TEST_P(surf, full, testing::Values(SURF_IMAGES))
{ {
String filename = getDataPath(GetParam()); String filename = getDataPath(GetParam());
Mat frame = imread(filename, IMREAD_GRAYSCALE); Mat frame = imread(filename, IMREAD_GRAYSCALE);
ASSERT_FALSE(frame.empty()) << "Unable to load source image " << filename;
if (frame.empty()) declare.in(frame).time(90);
FAIL() << "Unable to load source image " << filename;
Mat mask; Mat mask;
declare.in(frame).time(90); Ptr<Feature2D> detector;
Ptr<Feature2D> detector = getSURF();
vector<KeyPoint> points; vector<KeyPoint> points;
vector<float> descriptors; vector<float> descriptors;
TEST_CYCLE() detector->operator()(frame, mask, points, descriptors, false); if (getSelectedImpl() == "plain")
{
detector = new SURF;
TEST_CYCLE() detector->operator()(frame, mask, points, descriptors, false);
}
#ifdef HAVE_OPENCV_OCL
else if (getSelectedImpl() == "ocl")
{
detector = new ocl::SURF_OCL;
detector->operator()(frame, mask, points, noArray());
OCL_TEST_CYCLE() detector->operator()(frame, mask, points, descriptors, false);
}
#endif
else CV_TEST_FAIL_NO_IMPL();
SANITY_CHECK_KEYPOINTS(points, 1e-3); SANITY_CHECK_KEYPOINTS(points, 1e-3);
SANITY_CHECK(descriptors, 1e-4); SANITY_CHECK(descriptors, 1e-4);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment