cascade_haar_detector.h

//
// Projectname: amos-ss16-proj5
//
// Copyright (c) 2016 de.fau.cs.osr.amos2016.gruppe5
//
// This file is part of the AMOS Project 2016 @ FAU
// (Friedrich-Alexander University Erlangen-Nürnberg)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public
// License along with this program. If not, see
// <http://www.gnu.org/licenses/>.
//

#ifndef CASCADE_HAAR_DETECTOR_H
#define CASCADE_HAAR_DETECTOR_H

#include "detector.h"

class CascadeHaarDetector : public Detector {

public:

    CascadeHaarDetector(std::string classifier_file, double scale_factor = 1.1, int min_neighbors = 3, cv::Size min_size = cv::Size(20,20), cv::Size max_size = cv::Size(80,80));

    std::vector<cv::Rect> Detect(cv::Mat *frame);

    std::vector<cv::Rect> DetectInROI(cv::Mat *frame, std::vector<cv::Rect> *rois);

private:

    cv::CascadeClassifier cascade_haar_classifier_;
    double scale_factor_;
    int min_neighbors_;
    cv::Size min_size_;
    cv::Size max_size_;

};


#endif // CASCADE_HAAR_DETECTOR_H