diff options
author | Jörg Frings-Fürst <debian@jff.email> | 2018-10-01 07:17:16 +0200 |
---|---|---|
committer | Jörg Frings-Fürst <debian@jff.email> | 2018-10-01 07:17:16 +0200 |
commit | 71137cc5832110d91599d68790402cf196762ed3 (patch) | |
tree | e7985bbfb6ca50ea738b38a9f852ee51431d48a7 /facedetect/shotwell-facedetect.cpp | |
parent | 8ae550d463720d5d1c0e253af29843ff15de1a54 (diff) | |
parent | 1e92964463e564bb3359a7110342182fcfdc67f2 (diff) |
Merge branch 'release/debian/0.30.1-1'debian/0.30.1-1
Diffstat (limited to 'facedetect/shotwell-facedetect.cpp')
-rw-r--r-- | facedetect/shotwell-facedetect.cpp | 134 |
1 files changed, 134 insertions, 0 deletions
diff --git a/facedetect/shotwell-facedetect.cpp b/facedetect/shotwell-facedetect.cpp new file mode 100644 index 0000000..1b09379 --- /dev/null +++ b/facedetect/shotwell-facedetect.cpp @@ -0,0 +1,134 @@ +/* Copyright 2016 Software Freedom Conservancy Inc. + * + * Copyright 2011 Valentín Barros Puertas <valentin(at)sanva(dot)net> + * Copyright 2018 Ricardo Fantin da Costa <ricardofantin(at)gmail(dot)com> + * + * This software is licensed under the GNU LGPL (version 2.1 or later). + * See the COPYING file in this distribution. + */ + +#include "opencv2/objdetect/objdetect.hpp" +#include "opencv2/highgui/highgui.hpp" +#include "opencv2/imgproc/imgproc.hpp" + +#include <iostream> +#include <stdio.h> + +using namespace std; +using namespace cv; + +void help() { + + cout << + "Usage:" << endl << + "./facedetect --cascade=<cascade_path> " + "--scale=<image scale greater or equal to 1, try 1.3 for example> " + "filename" << endl << endl << + "Example:" << endl << + "./facedetect --cascade=\"./data/haarcascades/haarcascade_frontalface_alt.xml\" " + "--scale=1.3 ./photo.jpg" << endl << endl << + "Using OpenCV version " << CV_VERSION << endl; + +} + +void detectFaces(Mat &img, CascadeClassifier &cascade, double scale) { + + Mat gray; + cvtColor(img, gray, CV_BGR2GRAY); + + Mat smallImg(cvRound(img.rows / scale), cvRound(img.cols / scale), CV_8UC1); + Size smallImgSize = smallImg.size(); + + resize(gray, smallImg, smallImgSize, 0, 0, INTER_LINEAR); + equalizeHist(smallImg, smallImg); + + vector<Rect> faces; + cascade.detectMultiScale(smallImg, faces, 1.1, 2, CV_HAAR_SCALE_IMAGE, Size(30, 30)); + + int i = 0; + for (vector<Rect>::const_iterator r = faces.begin(); r != faces.end(); r++, i++) { + + printf( + "face;x=%f&y=%f&width=%f&height=%f\n", + (float) r->x / smallImgSize.width, + (float) r->y / smallImgSize.height, + (float) r->width / smallImgSize.width, + (float) r->height / smallImgSize.height + ); + + } + +} + +int main(int argc, const char** argv) { + + const std::string scaleOpt = "--scale="; + size_t scaleOptLen = scaleOpt.length(); + const std::string cascadeOpt = "--cascade="; + size_t cascadeOptLen = cascadeOpt.length(); + + std::string cascadeName, inputName; + double scale = 1; + + for (int i = 1; i < argc; i++) { + + if (cascadeOpt.compare(0, cascadeOptLen, argv[i], cascadeOptLen) == 0) { + + cascadeName.assign(argv[i] + cascadeOptLen); + + } else if (scaleOpt.compare(0, scaleOptLen, argv[i], scaleOptLen) == 0) { + + if (!sscanf(argv[i] + scaleOpt.length(), "%lf", &scale) || scale < 1) + scale = 1; + + } else if (argv[i][0] == '-') { + + cout << "warning;Unknown option " << argv[i] << endl; + + } else + inputName.assign(argv[i]); + + } + + if (cascadeName.empty()) { + + cout << "error;You must specify the cascade." << endl; + help(); + + return -1; + + } + + CascadeClassifier cascade; + + if (!cascade.load(cascadeName)) { + + cout << "error;Could not load classifier cascade. Filename: \"" << cascadeName << "\"" << endl; + + return -1; + } + + if (inputName.empty()) { + + cout << "error;You must specify the file to process." << endl; + help(); + + return -1; + + } + + Mat image = imread(inputName, 1); + + if (image.empty()) { + + cout << "error;Could not load the file to process. Filename: \"" << inputName << "\"" << endl; + + return -1; + + } + + detectFaces(image, cascade, scale); + + return 0; + +} |