mirror of
https://github.com/opencv/opencv_contrib.git
synced 2025-10-20 21:40:49 +08:00
c3a2b42552
Added if condition, whether faces are detected or not, will help to keep program running for some images. If imwrite used along with imshow, it may produce error for : No faces found.Aborting.. in function 'fit' (Bad Argument).
101 lines
3.6 KiB
C++
101 lines
3.6 KiB
C++
#include "opencv2/face.hpp"
|
|
#include "opencv2/videoio.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
#include "opencv2/imgcodecs.hpp"
|
|
#include "opencv2/objdetect.hpp"
|
|
#include "opencv2/imgproc.hpp"
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <string>
|
|
using namespace std;
|
|
using namespace cv;
|
|
using namespace cv::face;
|
|
|
|
static bool myDetector(InputArray image, OutputArray faces, CascadeClassifier *face_cascade)
|
|
{
|
|
Mat gray;
|
|
|
|
if (image.channels() > 1)
|
|
cvtColor(image, gray, COLOR_BGR2GRAY);
|
|
else
|
|
gray = image.getMat().clone();
|
|
|
|
equalizeHist(gray, gray);
|
|
|
|
std::vector<Rect> faces_;
|
|
face_cascade->detectMultiScale(gray, faces_, 1.4, 2, CASCADE_SCALE_IMAGE, Size(30, 30));
|
|
Mat(faces_).copyTo(faces);
|
|
return true;
|
|
}
|
|
|
|
int main(int argc,char** argv){
|
|
//Give the path to the directory containing all the files containing data
|
|
CommandLineParser parser(argc, argv,
|
|
"{ help h usage ? | | give the following arguments in following format }"
|
|
"{ model_filename f | | (required) path to binary file storing the trained model which is to be loaded [example - /data/file.dat]}"
|
|
"{ image i | | (required) path to image in which face landmarks have to be detected.[example - /data/image.jpg] }"
|
|
"{ face_cascade c | | Path to the face cascade xml file which you want to use as a detector}"
|
|
);
|
|
// Read in the input arguments
|
|
if (parser.has("help")){
|
|
parser.printMessage();
|
|
cerr << "TIP: Use absolute paths to avoid any problems with the software!" << endl;
|
|
return 0;
|
|
}
|
|
string filename(parser.get<string>("model_filename"));
|
|
if (filename.empty()){
|
|
parser.printMessage();
|
|
cerr << "The name of the model file to be loaded for detecting landmarks is not found" << endl;
|
|
return -1;
|
|
}
|
|
string image(parser.get<string>("image"));
|
|
if (image.empty()){
|
|
parser.printMessage();
|
|
cerr << "The name of the image file in which landmarks have to be detected is not found" << endl;
|
|
return -1;
|
|
}
|
|
string cascade_name(parser.get<string>("face_cascade"));
|
|
if (cascade_name.empty()){
|
|
parser.printMessage();
|
|
cerr << "The name of the cascade classifier to be loaded to detect faces is not found" << endl;
|
|
return -1;
|
|
}
|
|
|
|
Mat img = imread(image);
|
|
|
|
//pass the face cascade xml file which you want to pass as a detector
|
|
CascadeClassifier face_cascade;
|
|
face_cascade.load(cascade_name);
|
|
FacemarkKazemi::Params params;
|
|
Ptr<FacemarkKazemi> facemark = FacemarkKazemi::create(params);
|
|
facemark->setFaceDetector((FN_FaceDetector)myDetector, &face_cascade);
|
|
facemark->loadModel(filename);
|
|
cout<<"Loaded model"<<endl;
|
|
vector<Rect> faces;
|
|
resize(img,img,Size(460,460), 0, 0, INTER_LINEAR_EXACT);
|
|
facemark->getFaces(img,faces);
|
|
vector< vector<Point2f> > shapes;
|
|
|
|
// Check if faces detected or not
|
|
// Helps in proper exception handling when writing images to the directories.
|
|
if(faces.size() != 0) {
|
|
if(facemark->fit(img,faces,shapes))
|
|
{
|
|
for( size_t i = 0; i < faces.size(); i++ )
|
|
{
|
|
cv::rectangle(img,faces[i],Scalar( 255, 0, 0 ));
|
|
}
|
|
for(unsigned long i=0;i<faces.size();i++){
|
|
for(unsigned long k=0;k<shapes[i].size();k++)
|
|
cv::circle(img,shapes[i][k],5,cv::Scalar(0,0,255),FILLED);
|
|
}
|
|
namedWindow("Detected_shape");
|
|
imshow("Detected_shape",img);
|
|
waitKey(0);
|
|
}
|
|
} else {
|
|
cout << "Faces not detected." << endl;
|
|
}
|
|
return 0;
|
|
}
|