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

some fixes

Browse Source
This commit is contained in:
Konstantin Matskevich 2013-12-06 12:12:24 +04:00
parent 0a1ff0d42d
commit 7a78559708
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
samples/ocl/facedetect.cpp
View File

@ -74,7 +74,7 @@ static int facedetect_one_thread(bool useCPU, double scale )
if( !cascade.load( cascadeName ) || !cpu_cascade.load(cascadeName) ) if( !cascade.load( cascadeName ) || !cpu_cascade.load(cascadeName) )
{ {
cout << "ERROR: Could not load classifier cascade" << endl; cout << "ERROR: Could not load classifier cascade: " << cascadeName << endl;
return EXIT_FAILURE; return EXIT_FAILURE;
} }
@ -170,7 +170,12 @@ static int facedetect_one_thread(bool useCPU, double scale )
static void detectFaces(std::string fileName) static void detectFaces(std::string fileName)
{ {
ocl::OclCascadeClassifier cascade; ocl::OclCascadeClassifier cascade;
cascade.load(cascadeName); if(!cascade.load(cascadeName))
{
std::cout << "ERROR: Could not load classifier cascade: " << cascadeName << std::endl;
return;
}
Mat img = imread(fileName, CV_LOAD_IMAGE_COLOR); Mat img = imread(fileName, CV_LOAD_IMAGE_COLOR);
if (img.empty()) if (img.empty())
{ {
@ -187,8 +192,16 @@ static void detectFaces(std::string fileName)
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), Scalar( 0, 255, 255 ), 3); rectangle(img, Point(oclfaces[i].x, oclfaces[i].y), Point(oclfaces[i].x + oclfaces[i].width, oclfaces[i].y + oclfaces[i].height), Scalar( 0, 255, 255 ), 3);
imwrite(std::to_string(_threadid) + outputName, img); int n = (int)outputName.length();
while( n > 0 && outputName[n-1] != '.')
n--;
if( n == 0 )
{
std::cout << "Invalid output file name: " << outputName << std::endl;
return;
}
imwrite(outputName.substr(0,n-1) + "_" + std::to_string(_threadid) + outputName.substr(n-1, outputName.length()-1), img);
} }
static void facedetect_multithreading(int nthreads) static void facedetect_multithreading(int nthreads)