-
Notifications
You must be signed in to change notification settings - Fork 2
/
ObjectDetector.h
38 lines (30 loc) · 1023 Bytes
/
ObjectDetector.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#ifndef OBJECT_DETECTOR_H
#define OBJECT_DETECTOR_H
#include "Common.h"
#include "Detection.h"
#include "ParametersMap.h"
#include "SupportVectorMachine.h"
using namespace cv;
// Class responsible for performing Non-Maxima Suppression on the response
// pyramid generated by SupportVectorMachine
class ObjectDetector
{
public:
ObjectDetector(const SupportVectorMachine& svm);
~ObjectDetector();
void getDetections(Mat img, vector<Detection>& found);
void detect(const Mat& img, vector<Point>& hits, vector<double>& weights, double hitThreshold,
Size winStride, Size padding, const vector<Point>& locations, HOGDescriptor* hog);
private:
// HOGDescriptor _hog;
// vector<float> _svmDetector;
Size _winSize;
Size _blockSize;
Size _blockStride;
Size _cellSize;
int _nbins;
SupportVectorMachine _svm;
vector<float> _svmDetector;
void groupRectangles(vector<Rect>& rectList, vector<double>& weights, int groupThreshold, double eps);
};
#endif // OBJECT_DETECTOR_H