mirror of https://github.com/opencv/opencv.git
Merge pull request #1025 from bitwangyaoyao:2.4_tests
Roman Donchenko
12 years ago
committed by
OpenCV Buildbot
commit
4ed3d33dd7
21 changed files with 1307 additions and 1251 deletions
@ -1,180 +0,0 @@ |
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2010-2012, Institute Of Software Chinese Academy Of Science, all rights reserved.
|
||||
// Copyright (C) 2010-2012, Advanced Micro Devices, Inc., all rights reserved.
|
||||
// Copyright (C) 2010-2012, Multicoreware, Inc., all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// @Authors
|
||||
// Jia Haipeng, jiahaipeng95@gmail.com
|
||||
// Sen Liu, swjutls1987@126.com
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other oclMaterials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "opencv2/objdetect/objdetect.hpp" |
||||
#include "precomp.hpp" |
||||
|
||||
#ifdef HAVE_OPENCL |
||||
|
||||
using namespace cvtest; |
||||
using namespace testing; |
||||
using namespace std; |
||||
using namespace cv; |
||||
extern string workdir; |
||||
|
||||
namespace |
||||
{ |
||||
IMPLEMENT_PARAM_CLASS(CascadeName, std::string); |
||||
CascadeName cascade_frontalface_alt(std::string("haarcascade_frontalface_alt.xml")); |
||||
CascadeName cascade_frontalface_alt2(std::string("haarcascade_frontalface_alt2.xml")); |
||||
struct getRect |
||||
{ |
||||
Rect operator ()(const CvAvgComp &e) const |
||||
{ |
||||
return e.rect; |
||||
} |
||||
}; |
||||
} |
||||
|
||||
PARAM_TEST_CASE(Haar, double, int, CascadeName) |
||||
{ |
||||
cv::ocl::OclCascadeClassifier cascade, nestedCascade; |
||||
cv::CascadeClassifier cpucascade, cpunestedCascade; |
||||
|
||||
double scale; |
||||
int flags; |
||||
std::string cascadeName; |
||||
|
||||
virtual void SetUp() |
||||
{ |
||||
scale = GET_PARAM(0); |
||||
flags = GET_PARAM(1); |
||||
cascadeName = (workdir + "../../data/haarcascades/").append(GET_PARAM(2)); |
||||
|
||||
if( (!cascade.load( cascadeName )) || (!cpucascade.load(cascadeName)) ) |
||||
{ |
||||
cout << "ERROR: Could not load classifier cascade" << endl; |
||||
return; |
||||
} |
||||
} |
||||
}; |
||||
|
||||
////////////////////////////////faceDetect/////////////////////////////////////////////////
|
||||
TEST_P(Haar, FaceDetect) |
||||
{ |
||||
string imgName = workdir + "lena.jpg"; |
||||
Mat img = imread( imgName, 1 ); |
||||
|
||||
if(img.empty()) |
||||
{ |
||||
std::cout << "Couldn't read " << imgName << std::endl; |
||||
return ; |
||||
} |
||||
|
||||
vector<Rect> faces, oclfaces; |
||||
|
||||
Mat gray, smallImg(cvRound (img.rows / scale), cvRound(img.cols / scale), CV_8UC1 ); |
||||
MemStorage storage(cvCreateMemStorage(0)); |
||||
cvtColor( img, gray, CV_BGR2GRAY ); |
||||
resize( gray, smallImg, smallImg.size(), 0, 0, INTER_LINEAR ); |
||||
equalizeHist( smallImg, smallImg ); |
||||
|
||||
cv::ocl::oclMat image; |
||||
CvSeq *_objects; |
||||
image.upload(smallImg); |
||||
_objects = cascade.oclHaarDetectObjects( image, storage, 1.1, |
||||
3, flags, Size(30, 30), Size(0, 0) ); |
||||
vector<CvAvgComp> vecAvgComp; |
||||
Seq<CvAvgComp>(_objects).copyTo(vecAvgComp); |
||||
oclfaces.resize(vecAvgComp.size()); |
||||
std::transform(vecAvgComp.begin(), vecAvgComp.end(), oclfaces.begin(), getRect()); |
||||
|
||||
cpucascade.detectMultiScale( smallImg, faces, 1.1, 3, |
||||
flags, |
||||
Size(30, 30), Size(0, 0) ); |
||||
EXPECT_EQ(faces.size(), oclfaces.size()); |
||||
} |
||||
|
||||
TEST_P(Haar, FaceDetectUseBuf) |
||||
{ |
||||
string imgName = workdir + "lena.jpg"; |
||||
Mat img = imread( imgName, 1 ); |
||||
|
||||
if(img.empty()) |
||||
{ |
||||
std::cout << "Couldn't read " << imgName << std::endl; |
||||
return ; |
||||
} |
||||
|
||||
vector<Rect> faces, oclfaces; |
||||
|
||||
Mat gray, smallImg(cvRound (img.rows / scale), cvRound(img.cols / scale), CV_8UC1 ); |
||||
cvtColor( img, gray, CV_BGR2GRAY ); |
||||
resize( gray, smallImg, smallImg.size(), 0, 0, INTER_LINEAR ); |
||||
equalizeHist( smallImg, smallImg ); |
||||
|
||||
cv::ocl::oclMat image; |
||||
image.upload(smallImg); |
||||
|
||||
cv::ocl::OclCascadeClassifierBuf cascadebuf; |
||||
if( !cascadebuf.load( cascadeName ) ) |
||||
{ |
||||
cout << "ERROR: Could not load classifier cascade for FaceDetectUseBuf!" << endl; |
||||
return; |
||||
} |
||||
cascadebuf.detectMultiScale( image, oclfaces, 1.1, 3, |
||||
flags, |
||||
Size(30, 30), Size(0, 0) ); |
||||
|
||||
cpucascade.detectMultiScale( smallImg, faces, 1.1, 3, |
||||
flags, |
||||
Size(30, 30), Size(0, 0) ); |
||||
EXPECT_EQ(faces.size(), oclfaces.size()); |
||||
|
||||
// intentionally run ocl facedetect again and check if it still works after the first run
|
||||
cascadebuf.detectMultiScale( image, oclfaces, 1.1, 3, |
||||
flags, |
||||
Size(30, 30)); |
||||
cascadebuf.release(); |
||||
EXPECT_EQ(faces.size(), oclfaces.size()); |
||||
} |
||||
|
||||
INSTANTIATE_TEST_CASE_P(FaceDetect, Haar, |
||||
Combine(Values(1.0), |
||||
Values(CV_HAAR_SCALE_IMAGE, 0), Values(cascade_frontalface_alt, cascade_frontalface_alt2))); |
||||
|
||||
#endif // HAVE_OPENCL
|
||||
@ -1,91 +0,0 @@ |
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2010-2012, Multicoreware, Inc., all rights reserved.
|
||||
// Copyright (C) 2010-2012, Advanced Micro Devices, Inc., all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// @Authors
|
||||
// Zhang Chunpeng chunpeng@multicorewareinc.com
|
||||
// Yao Wang yao@multicorewareinc.com
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other oclMaterials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "precomp.hpp" |
||||
#include "opencv2/core/core.hpp" |
||||
|
||||
#ifdef HAVE_OPENCL |
||||
|
||||
using namespace cv; |
||||
using namespace cvtest; |
||||
using namespace testing; |
||||
using namespace std; |
||||
|
||||
PARAM_TEST_CASE(PyrUp, MatType, int) |
||||
{ |
||||
int type; |
||||
int channels; |
||||
|
||||
virtual void SetUp() |
||||
{ |
||||
type = GET_PARAM(0); |
||||
channels = GET_PARAM(1); |
||||
} |
||||
}; |
||||
|
||||
TEST_P(PyrUp, Accuracy) |
||||
{ |
||||
for(int j = 0; j < LOOP_TIMES; j++) |
||||
{ |
||||
Size size(MWIDTH, MHEIGHT); |
||||
Mat src = randomMat(size, CV_MAKETYPE(type, channels)); |
||||
Mat dst_gold; |
||||
pyrUp(src, dst_gold); |
||||
ocl::oclMat dst; |
||||
ocl::oclMat srcMat(src); |
||||
ocl::pyrUp(srcMat, dst); |
||||
|
||||
EXPECT_MAT_NEAR(dst_gold, Mat(dst), (type == CV_32F ? 1e-4f : 1.0)); |
||||
} |
||||
|
||||
} |
||||
|
||||
|
||||
INSTANTIATE_TEST_CASE_P(OCL_ImgProc, PyrUp, testing::Combine( |
||||
Values(CV_8U, CV_32F), Values(1, 3, 4))); |
||||
|
||||
|
||||
#endif // HAVE_OPENCL
|
||||
@ -0,0 +1,265 @@ |
||||
#include <iostream> |
||||
#include <vector> |
||||
#include <iomanip> |
||||
|
||||
#include "opencv2/highgui/highgui.hpp" |
||||
#include "opencv2/ocl/ocl.hpp" |
||||
#include "opencv2/video/video.hpp" |
||||
|
||||
using namespace std; |
||||
using namespace cv; |
||||
using namespace cv::ocl; |
||||
|
||||
typedef unsigned char uchar; |
||||
#define LOOP_NUM 10 |
||||
int64 work_begin = 0; |
||||
int64 work_end = 0; |
||||
|
||||
static void workBegin() |
||||
{ |
||||
work_begin = getTickCount(); |
||||
} |
||||
static void workEnd() |
||||
{ |
||||
work_end += (getTickCount() - work_begin); |
||||
} |
||||
static double getTime() |
||||
{ |
||||
return work_end * 1000. / getTickFrequency(); |
||||
} |
||||
|
||||
template <typename T> inline T clamp (T x, T a, T b) |
||||
{ |
||||
return ((x) > (a) ? ((x) < (b) ? (x) : (b)) : (a)); |
||||
} |
||||
|
||||
template <typename T> inline T mapValue(T x, T a, T b, T c, T d) |
||||
{ |
||||
x = clamp(x, a, b); |
||||
return c + (d - c) * (x - a) / (b - a); |
||||
} |
||||
|
||||
static void getFlowField(const Mat& u, const Mat& v, Mat& flowField) |
||||
{ |
||||
float maxDisplacement = 1.0f; |
||||
|
||||
for (int i = 0; i < u.rows; ++i) |
||||
{ |
||||
const float* ptr_u = u.ptr<float>(i); |
||||
const float* ptr_v = v.ptr<float>(i); |
||||
|
||||
for (int j = 0; j < u.cols; ++j) |
||||
{ |
||||
float d = max(fabsf(ptr_u[j]), fabsf(ptr_v[j])); |
||||
|
||||
if (d > maxDisplacement) |
||||
maxDisplacement = d; |
||||
} |
||||
} |
||||
|
||||
flowField.create(u.size(), CV_8UC4); |
||||
|
||||
for (int i = 0; i < flowField.rows; ++i) |
||||
{ |
||||
const float* ptr_u = u.ptr<float>(i); |
||||
const float* ptr_v = v.ptr<float>(i); |
||||
|
||||
|
||||
Vec4b* row = flowField.ptr<Vec4b>(i); |
||||
|
||||
for (int j = 0; j < flowField.cols; ++j) |
||||
{ |
||||
row[j][0] = 0; |
||||
row[j][1] = static_cast<unsigned char> (mapValue (-ptr_v[j], -maxDisplacement, maxDisplacement, 0.0f, 255.0f)); |
||||
row[j][2] = static_cast<unsigned char> (mapValue ( ptr_u[j], -maxDisplacement, maxDisplacement, 0.0f, 255.0f)); |
||||
row[j][3] = 255; |
||||
} |
||||
} |
||||
} |
||||
|
||||
|
||||
int main(int argc, const char* argv[]) |
||||
{ |
||||
static std::vector<Info> ocl_info; |
||||
ocl::getDevice(ocl_info); |
||||
//if you want to use undefault device, set it here
|
||||
setDevice(ocl_info[0]); |
||||
|
||||
//set this to save kernel compile time from second time you run
|
||||
ocl::setBinpath("./"); |
||||
const char* keys = |
||||
"{ h | help | false | print help message }" |
||||
"{ l | left | | specify left image }" |
||||
"{ r | right | | specify right image }" |
||||
"{ o | output | tvl1_output.jpg | specify output save path }" |
||||
"{ c | camera | 0 | enable camera capturing }" |
||||
"{ s | use_cpu | false | use cpu or gpu to process the image }" |
||||
"{ v | video | | use video as input }"; |
||||
|
||||
CommandLineParser cmd(argc, argv, keys); |
||||
|
||||
if (cmd.get<bool>("help")) |
||||
{ |
||||
cout << "Usage: pyrlk_optical_flow [options]" << endl; |
||||
cout << "Avaible options:" << endl; |
||||
cmd.printParams(); |
||||
return 0; |
||||
} |
||||
|
||||
bool defaultPicturesFail = false; |
||||
string fname0 = cmd.get<string>("l"); |
||||
string fname1 = cmd.get<string>("r"); |
||||
string vdofile = cmd.get<string>("v"); |
||||
string outpath = cmd.get<string>("o"); |
||||
bool useCPU = cmd.get<bool>("s"); |
||||
bool useCamera = cmd.get<bool>("c"); |
||||
int inputName = cmd.get<int>("c"); |
||||
|
||||
Mat frame0 = imread(fname0, cv::IMREAD_GRAYSCALE); |
||||
Mat frame1 = imread(fname1, cv::IMREAD_GRAYSCALE); |
||||
cv::Ptr<cv::DenseOpticalFlow> alg = cv::createOptFlow_DualTVL1(); |
||||
cv::ocl::OpticalFlowDual_TVL1_OCL d_alg; |
||||
|
||||
|
||||
Mat flow, show_flow; |
||||
Mat flow_vec[2]; |
||||
if (frame0.empty() || frame1.empty()) |
||||
{ |
||||
useCamera = true; |
||||
defaultPicturesFail = true; |
||||
CvCapture* capture = 0; |
||||
capture = cvCaptureFromCAM( inputName ); |
||||
if (!capture) |
||||
{ |
||||
cout << "Can't load input images" << endl; |
||||
return -1; |
||||
} |
||||
} |
||||
|
||||
|
||||
if (useCamera) |
||||
{ |
||||
CvCapture* capture = 0; |
||||
Mat frame, frameCopy; |
||||
Mat frame0Gray, frame1Gray; |
||||
Mat ptr0, ptr1; |
||||
|
||||
if(vdofile == "") |
||||
capture = cvCaptureFromCAM( inputName ); |
||||
else |
||||
capture = cvCreateFileCapture(vdofile.c_str()); |
||||
|
||||
int c = inputName ; |
||||
if(!capture) |
||||
{ |
||||
if(vdofile == "") |
||||
cout << "Capture from CAM " << c << " didn't work" << endl; |
||||
else |
||||
cout << "Capture from file " << vdofile << " failed" <<endl; |
||||
if (defaultPicturesFail) |
||||
{ |
||||
return -1; |
||||
} |
||||
goto nocamera; |
||||
} |
||||
|
||||
cout << "In capture ..." << endl; |
||||
for(int i = 0;; i++) |
||||
{ |
||||
frame = cvQueryFrame( capture ); |
||||
if( frame.empty() ) |
||||
break; |
||||
|
||||
if (i == 0) |
||||
{ |
||||
frame.copyTo( frame0 ); |
||||
cvtColor(frame0, frame0Gray, COLOR_BGR2GRAY); |
||||
} |
||||
else |
||||
{ |
||||
if (i%2 == 1) |
||||
{ |
||||
frame.copyTo(frame1); |
||||
cvtColor(frame1, frame1Gray, COLOR_BGR2GRAY); |
||||
ptr0 = frame0Gray; |
||||
ptr1 = frame1Gray; |
||||
} |
||||
else |
||||
{ |
||||
frame.copyTo(frame0); |
||||
cvtColor(frame0, frame0Gray, COLOR_BGR2GRAY); |
||||
ptr0 = frame1Gray; |
||||
ptr1 = frame0Gray; |
||||
} |
||||
|
||||
if (useCPU) |
||||
{ |
||||
alg->calc(ptr0, ptr1, flow); |
||||
split(flow, flow_vec); |
||||
} |
||||
else |
||||
{ |
||||
oclMat d_flowx, d_flowy; |
||||
d_alg(oclMat(ptr0), oclMat(ptr1), d_flowx, d_flowy); |
||||
d_flowx.download(flow_vec[0]); |
||||
d_flowy.download(flow_vec[1]); |
||||
} |
||||
if (i%2 == 1) |
||||
frame1.copyTo(frameCopy); |
||||
else |
||||
frame0.copyTo(frameCopy); |
||||
getFlowField(flow_vec[0], flow_vec[1], show_flow); |
||||
imshow("PyrLK [Sparse]", show_flow); |
||||
} |
||||
|
||||
if( waitKey( 10 ) >= 0 ) |
||||
goto _cleanup_; |
||||
} |
||||
|
||||
waitKey(0); |
||||
|
||||
_cleanup_: |
||||
cvReleaseCapture( &capture ); |
||||
} |
||||
else |
||||
{ |
||||
nocamera: |
||||
oclMat d_flowx, d_flowy; |
||||
for(int i = 0; i <= LOOP_NUM; i ++) |
||||
{ |
||||
cout << "loop" << i << endl; |
||||
|
||||
if (i > 0) workBegin(); |
||||
if (useCPU) |
||||
{ |
||||
alg->calc(frame0, frame1, flow); |
||||
split(flow, flow_vec); |
||||
} |
||||
else |
||||
{ |
||||
d_alg(oclMat(frame0), oclMat(frame1), d_flowx, d_flowy); |
||||
d_flowx.download(flow_vec[0]); |
||||
d_flowy.download(flow_vec[1]); |
||||
} |
||||
if (i > 0 && i <= LOOP_NUM) |
||||
workEnd(); |
||||
|
||||
if (i == LOOP_NUM) |
||||
{ |
||||
if (useCPU) |
||||
cout << "average CPU time (noCamera) : "; |
||||
else |
||||
cout << "average GPU time (noCamera) : "; |
||||
cout << getTime() / LOOP_NUM << " ms" << endl; |
||||
|
||||
getFlowField(flow_vec[0], flow_vec[1], show_flow); |
||||
imshow("PyrLK [Sparse]", show_flow); |
||||
imwrite(outpath, show_flow); |
||||
} |
||||
} |
||||
} |
||||
|
||||
waitKey(); |
||||
|
||||
return 0; |
||||
} |
||||
Loading…
Reference in new issue