PedestrainCounting.h

/*******************************************
 Pedestrain Count System.
 @author Zhengrong Wang
 *******************************************/

#ifndef PEDESTRAIN_COUNTING_HEADER
#define PEDESTRAIN_COUNTING_HEADER

#include "kmeans.h"
#include "meanshift.h"

#include "VideoDetector.h"
#include "BKGCutDetector.h"
#include "AdaBoostClassifier.h"
#include "HoGIntegralImage.h"
#include "RGIIntegralImage.h"
#include "GrayScaleIntegralImage.h"
#include "StrongClassifierDirectSelect.h"
#include "ParticleFilterTracker.h"
#include "MultiTracker.h"

const int BKG_DETECTION_NUM_CLUSTER = 3;
const int BKG_VIDEO_PIC_NUM_ALL = 50;

#define adaboostFile "HOGClassifier\\"

#define HEIGHT 64
#define WIDTH 32

class PedestrainCounter {
public:

    PedestrainCounter();
    ~PedestrainCounter() {}

    void ParseParams(int argc, char *argv[]);

    void PrintHelp();

    void DetectSinglePictureHOG(const char *in, const char *out,
        const char *adaboost, const Options &opt);

    void DetectSinglePictureBKG(const char *in, const char *bkg, const char *out,
        const char *adaboost, const Options &opt);

    void DetectVideoHOG(const char *in, const char *out,
        const char *adaboost, const Options &opt);

    void DetectVideoBKG(const char *in, const char *bkg, const char *out,
        const char *adaboost, const Options &opt);

    void TrackVideoSingle(const char *in, const char *out);

    void TrackVideoMulti(const char *in, const char *bkg, const char *out,
        const char *adaboost, const Options &opt);

    static void GetTarget(int event, int x, int y, int flags, void *userParams);

    IplImage* video_bkg_detect(
        const TCHAR *pIn, const TCHAR* pTemp, int num_cluster, int r, int pic_num_all,
        int H_thres, int W_thres);

    void create_meanshift_queue(const TCHAR *pIn,
        const TCHAR* pOut, int r, int pic_num_all);

    void readin_tuples(
        TCHAR pIn[], Tuple* &tuples, int pic_num_all, CvRect roi,
        int frameH, int frameW);

private:
    Options opt;
};

#endif