parent
a8f9344ef8
commit
6f138e50c2
1 changed files with 62 additions and 0 deletions
@ -0,0 +1,62 @@ |
||||
#ifndef __OPENCV_XOBJDETECT_PRIVATE_HPP__ |
||||
#define __OPENCV_XOBJDETECT_PRIVATE_HPP__ |
||||
|
||||
#ifndef __OPENCV_BUILD |
||||
# error this is a private header, do not include it outside OpenCV |
||||
#endif |
||||
|
||||
#include <opencv2/core.hpp> |
||||
|
||||
namespace cv |
||||
{ |
||||
namespace xobjdetect |
||||
{ |
||||
|
||||
class CV_EXPORTS Stump |
||||
{ |
||||
public: |
||||
|
||||
/* Initialize zero stump */ |
||||
Stump(): threshold_(0), polarity_(1), pos_value_(1), neg_value_(-1) {} |
||||
|
||||
/* Initialize stump with given threshold, polarity
|
||||
and classification values */ |
||||
Stump(int threshold, int polarity, float pos_value, float neg_value): |
||||
threshold_(threshold), polarity_(polarity), |
||||
pos_value_(pos_value), neg_value_(neg_value) {} |
||||
|
||||
/* Train stump for given data
|
||||
|
||||
data — matrix of feature values, size M x N, one feature per row |
||||
|
||||
labels — matrix of sample class labels, size 1 x N. Labels can be from |
||||
{-1, +1} |
||||
|
||||
weights — matrix of sample weights, size 1 x N |
||||
|
||||
Returns chosen feature index. Feature enumeration starts from 0 |
||||
*/ |
||||
int train(const Mat& data, const Mat& labels, const Mat& weights); |
||||
|
||||
/* Predict object class given
|
||||
|
||||
value — feature value. Feature must be the same as was chosen |
||||
during training stump |
||||
|
||||
Returns real value, sign(value) means class
|
||||
*/ |
||||
float predict(int value) const; |
||||
|
||||
private: |
||||
/* Stump decision threshold */ |
||||
int threshold_; |
||||
/* Stump polarity, can be from {-1, +1} */ |
||||
int polarity_; |
||||
/* Classification values for positive and negative classes */ |
||||
float pos_value_, neg_value_; |
||||
}; |
||||
|
||||
} /* namespace xobjdetect */ |
||||
} /* namespace cv */ |
||||
|
||||
#endif // __OPENCV_XOBJDETECT_PRIVATE_HPP__
|
Loading…
Reference in new issue