generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2013 from apavlenko:2.4_fd_sample

Browse Source
This commit is contained in:
Andrey Pavlenko 2013-12-18 17:05:17 +04:00 committed by OpenCV Buildbot
commit ae9d24d433
1 changed files with 55 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
samples/ocl/facedetect.cpp
View File

@ -11,7 +11,10 @@
using namespace std; using namespace std;
using namespace cv; using namespace cv;
#define LOOP_NUM 10 #define LOOP_NUM 10
#define MAX_THREADS 10
///////////////////////////single-threading faces detecting/////////////////////////////// ///////////////////////////single-threading faces detecting///////////////////////////////
@ -26,23 +29,23 @@ const static Scalar colors[] = { CV_RGB(0,0,255),
} ; } ;
int64 work_begin = 0; int64 work_begin[MAX_THREADS] = {0};
int64 work_end = 0; int64 work_total[MAX_THREADS] = {0};
string inputName, outputName, cascadeName; string inputName, outputName, cascadeName;
static void workBegin() static void workBegin(int i = 0)
{ {
work_begin = getTickCount(); work_begin[i] = getTickCount();
} }
static void workEnd() static void workEnd(int i = 0)
{ {
work_end += (getTickCount() - work_begin); work_total[i] += (getTickCount() - work_begin[i]);
} }
static double getTime() static double getTotalTime(int i = 0)
{ {
return work_end /((double)cvGetTickFrequency() * 1000.); return work_total[i] /getTickFrequency() * 1000.;
} }
@ -96,7 +99,6 @@ static int facedetect_one_thread(bool useCPU, double scale )
} }
} }
cvNamedWindow( "result", 1 );
if( capture ) if( capture )
{ {
cout << "In capture ..." << endl; cout << "In capture ..." << endl;
@ -125,34 +127,34 @@ static int facedetect_one_thread(bool useCPU, double scale )
} }
else else
{ {
cout << "In image read" << endl; cout << "In image read " << image.size() << endl;
vector<Rect> faces; vector<Rect> faces;
vector<Rect> ref_rst; vector<Rect> ref_rst;
double accuracy = 0.; double accuracy = 0.;
for(int i = 0; i <= LOOP_NUM; i ++) cout << "loops: ";
for(int i = 0; i <= LOOP_NUM; i++)
{ {
cout << "loop" << i << endl; cout << i << ", ";
if(useCPU) if(useCPU)
detectCPU(image, faces, cpu_cascade, scale, i==0?false:true); detectCPU(image, faces, cpu_cascade, scale, i!=0);
else else
{ {
detect(image, faces, cascade, scale, i==0?false:true); detect(image, faces, cascade, scale, i!=0);
if(i == 0) if(i == 0)
{ {
detectCPU(image, ref_rst, cpu_cascade, scale, false); detectCPU(image, ref_rst, cpu_cascade, scale, false);
accuracy = checkRectSimilarity(image.size(), ref_rst, faces); accuracy = checkRectSimilarity(image.size(), ref_rst, faces);
} }
} }
if (i == LOOP_NUM)
{
if (useCPU)
cout << "average CPU time (noCamera) : ";
else
cout << "average GPU time (noCamera) : ";
cout << getTime() / LOOP_NUM << " ms" << endl;
cout << "accuracy value: " << accuracy <<endl;
}
} }
cout << "done!" << endl;
if (useCPU)
cout << "average CPU time (noCamera) : ";
else
cout << "average GPU time (noCamera) : ";
cout << getTotalTime() / LOOP_NUM << " ms" << endl;
cout << "accuracy value: " << accuracy <<endl;
Draw(image, faces, scale); Draw(image, faces, scale);
waitKey(0); waitKey(0);
} }
@ -165,9 +167,7 @@ static int facedetect_one_thread(bool useCPU, double scale )
///////////////////////////////////////detectfaces with multithreading//////////////////////////////////////////// ///////////////////////////////////////detectfaces with multithreading////////////////////////////////////////////
#if defined(_MSC_VER) && (_MSC_VER >= 1700) #if defined(_MSC_VER) && (_MSC_VER >= 1700)
#define MAX_THREADS 10 static void detectFaces(std::string fileName, int threadNum)
static void detectFaces(std::string fileName)
{ {
ocl::OclCascadeClassifier cascade; ocl::OclCascadeClassifier cascade;
if(!cascade.load(cascadeName)) if(!cascade.load(cascadeName))
@ -179,7 +179,7 @@ static void detectFaces(std::string fileName)
Mat img = imread(fileName, CV_LOAD_IMAGE_COLOR); Mat img = imread(fileName, CV_LOAD_IMAGE_COLOR);
if (img.empty()) if (img.empty())
{ {
std::cout << "cann't open file " + fileName <<std::endl; std::cout << '[' << threadNum << "] " << "can't open file " + fileName <<std::endl;
return; return;
} }
@ -187,23 +187,37 @@ static void detectFaces(std::string fileName)
d_img.upload(img); d_img.upload(img);
std::vector<Rect> oclfaces; std::vector<Rect> oclfaces;
cascade.detectMultiScale(d_img, oclfaces, 1.1, 3, 0|CV_HAAR_SCALE_IMAGE, Size(30, 30), Size(0, 0)); std::thread::id tid = std::this_thread::get_id();
std::cout << '[' << threadNum << "] "
<< "ThreadID = " << tid
<< ", CommandQueue = " << *(void**)ocl::getClCommandQueuePtr()
<< endl;
for(int i = 0; i <= LOOP_NUM; i++)
{
if(i>0) workBegin(threadNum);
cascade.detectMultiScale(d_img, oclfaces, 1.1, 3, 0|CV_HAAR_SCALE_IMAGE, Size(30, 30), Size(0, 0));
if(i>0) workEnd(threadNum);
}
std::cout << '[' << threadNum << "] " << "Average time = " << getTotalTime(threadNum) / LOOP_NUM << " ms" << endl;
for(unsigned int i = 0; i<oclfaces.size(); i++) for(unsigned int i = 0; i<oclfaces.size(); i++)
rectangle(img, Point(oclfaces[i].x, oclfaces[i].y), Point(oclfaces[i].x + oclfaces[i].width, oclfaces[i].y + oclfaces[i].height), colors[i%8], 3); rectangle(img, Point(oclfaces[i].x, oclfaces[i].y), Point(oclfaces[i].x + oclfaces[i].width, oclfaces[i].y + oclfaces[i].height), colors[i%8], 3);
std::string::size_type pos = outputName.rfind('.'); std::string::size_type pos = outputName.rfind('.');
std::string outputNameTid = outputName + '-' + std::to_string(_threadid); std::string strTid = std::to_string(_threadid);
if(pos == std::string::npos) if( !outputName.empty() )
{ {
std::cout << "Invalid output file name: " << outputName << std::endl; if(pos == std::string::npos)
{
std::cout << "Invalid output file name: " << outputName << std::endl;
}
else
{
std::string outputNameTid = outputName.substr(0, pos) + "_" + strTid + outputName.substr(pos);
imwrite(outputNameTid, img);
}
} }
else imshow(strTid, img);
{
outputNameTid = outputName.substr(0, pos) + "_" + std::to_string(_threadid) + outputName.substr(pos);
imwrite(outputNameTid, img);
}
imshow(outputNameTid, img);
waitKey(0); waitKey(0);
} }
@ -212,7 +226,7 @@ static void facedetect_multithreading(int nthreads)
int thread_number = MAX_THREADS < nthreads ? MAX_THREADS : nthreads; int thread_number = MAX_THREADS < nthreads ? MAX_THREADS : nthreads;
std::vector<std::thread> threads; std::vector<std::thread> threads;
for(int i = 0; i<thread_number; i++) for(int i = 0; i<thread_number; i++)
threads.push_back(std::thread(detectFaces, inputName)); threads.push_back(std::thread(detectFaces, inputName, i));
for(int i = 0; i<thread_number; i++) for(int i = 0; i<thread_number; i++)
threads[i].join(); threads[i].join();
} }
@ -228,8 +242,7 @@ int main( int argc, const char** argv )
" specify template file path }" " specify template file path }"
"{ c | scale | 1.0 | scale image }" "{ c | scale | 1.0 | scale image }"
"{ s | use_cpu | false | use cpu or gpu to process the image }" "{ s | use_cpu | false | use cpu or gpu to process the image }"
"{ o | output | facedetect_output.jpg |" "{ o | output | | specify output image save path(only works when input is images) }"
" specify output image save path(only works when input is images) }"
"{ n | thread_num | 1 | set number of threads >= 1 }"; "{ n | thread_num | 1 | set number of threads >= 1 }";
CommandLineParser cmd(argc, argv, keys); CommandLineParser cmd(argc, argv, keys);
@ -314,8 +327,8 @@ void Draw(Mat& img, vector<Rect>& faces, double scale)
radius = cvRound((r->width + r->height)*0.25*scale); radius = cvRound((r->width + r->height)*0.25*scale);
circle( img, center, radius, color, 3, 8, 0 ); circle( img, center, radius, color, 3, 8, 0 );
} }
imwrite( outputName, img ); if( !outputName.empty() ) imwrite( outputName, img );
if(abs(scale-1.0)>.001) if( abs(scale-1.0)>.001 )
{ {
resize(img, img, Size((int)(img.cols/scale), (int)(img.rows/scale))); resize(img, img, Size((int)(img.cols/scale), (int)(img.rows/scale)));
} }