ImageDetector.h

#ifndef IMAGE_DETECTOR_HEADER
#define IMAGE_DETECTOR_HEADER

// #include "FeatureExtractor.h"
#include "Options.h"
#include "Classifier.h"
#include "Pool.h"
#include "UnionFind.h"

struct rect
{
    int x1;
    int y1;
    int x2;
    int y2;
    int re;
    int count;
    rect(int _x1 = 0, int _y1 = 0, int _x2 = 0, int _y2 = 0)
        : x1(_x1), y1(_y1), x2(_x2), y2(_y2), re(-1), count(1) {}
};

// Use some specific feature and classifier to detect pedestrain.
// Apply this directly on the whole image.
class ImageDetector {
public:
    ImageDetector(Classifier *c, const Options &op);
    
    virtual bool Detect(const cv::Mat &img, const IntegralImage *intImage,
        const Rect &subRegion,
        const cv::Mat &bkg = defaultBackground
        );

    void DrawDetection(cv::Mat &img);

    // Clear all the detections.
    void Clear();

    // Detections.
    Pool<Rect> dets;

protected:
    Classifier *classifier;
    feat scaleMin;
    feat scaleMax;
    feat scaleStep;
    int slideStep;
    int evidence;
    int modelHeight;
    int modelWidth;

    
    // Temp pool used in merge.
    Pool<rect> temp;

    // This is used in union-find.
    Pool<int> roots;

    UnionFind *unionFind;

    // Some helper functions.
    // Is these two detection the same one?
    bool IsEqual(const rect &r1, const rect &r2) const;
    bool IsOverlap(const rect &r, const rect &t) const;
};

void DrawDetection(cv::Mat &img, Pool<rect> &dets);


#endif