cv_object_detect.h

Go to the documentation of this file.

/* =========================================================================== *
   |
   |  Copyright (c) 1994-2011 by Kobus Barnard (author)
   |
   |  Personal and educational use of this code is granted, provided that this
   |  header is kept intact, and that the authorship is not misrepresented, that
   |  its use is acknowledged in publications, and relevant papers are cited.
   |
   |  For other use contact the author (kobus AT cs DOT arizona DOT edu).
   |
   |  Please note that the code in this file has not necessarily been adequately
   |  tested. Naturally, there is no guarantee of performance, support, or fitness
   |  for any particular task. Nonetheless, I am interested in hearing about
   |  problems that you encounter.
   |
   |  Author:  Jinyan Guan
 * =========================================================================== */

/* $Id: cv_object_detect.h 17393 2014-08-23 20:19:14Z predoehl $ */

#ifndef CV_OBJECT_DETECT_H_
#define CV_OBJECT_DEFECT_H_

#ifdef KJB_HAVE_OPENCV
#include <wrap_opencv_cpp/cv.h>
#endif

#include <gr_cpp/gr_2D_bounding_box.h>
#include <wrap_opencv_cpp/cv_util.h>

#include <vector>

namespace kjb
{
namespace opencv
{
#ifdef KJB_HAVE_OPENCV

class CV_cascade_classifier
{
public:
    // Constructors ... 
    CV_cascade_classifier(){}

    CV_cascade_classifier(const std::string& classifier_fname)
        : cv_classifier(classifier_fname)
    {}

    ~CV_cascade_classifier(){}

public:
    bool load(const std::string& classifier_fname)
    {
        return cv_classifier.load(classifier_fname);
    }

    void detect_multiscale
    (
        const Image& img,
        std::vector<Axis_aligned_rectangle_2d>& objs, 
        double scale_factor = 1.1,
        int min_neighbors = 3,
        int flags = 0, 
        int width = 0,
        int height = 0
    );

private:
    cv::CascadeClassifier cv_classifier;

}; // class CV_cascade_classifier

#endif

} //namespace cv
} //namespace kjb
#endif /* CV_OBJECT_DETECT_H_ */