mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.6 KiB
78 lines
2.6 KiB
#ifndef _OPENCV_HOGFEATURES_H_ |
|
#define _OPENCV_HOGFEATURES_H_ |
|
|
|
#include "traincascade_features.h" |
|
|
|
//#define TEST_INTHIST_BUILD |
|
//#define TEST_FEAT_CALC |
|
|
|
#define N_BINS 9 |
|
#define N_CELLS 4 |
|
|
|
#define HOGF_NAME "HOGFeatureParams" |
|
struct CvHOGFeatureParams : public CvFeatureParams |
|
{ |
|
CvHOGFeatureParams(); |
|
}; |
|
|
|
class CvHOGEvaluator : public CvFeatureEvaluator |
|
{ |
|
public: |
|
virtual ~CvHOGEvaluator() {} |
|
virtual void init(const CvFeatureParams *_featureParams, |
|
int _maxSampleCount, cv::Size _winSize ); |
|
virtual void setImage(const cv::Mat& img, uchar clsLabel, int idx); |
|
virtual float operator()(int varIdx, int sampleIdx) const; |
|
virtual void writeFeatures( cv::FileStorage &fs, const cv::Mat& featureMap ) const; |
|
protected: |
|
virtual void generateFeatures(); |
|
virtual void integralHistogram(const cv::Mat &img, std::vector<cv::Mat> &histogram, cv::Mat &norm, int nbins) const; |
|
class Feature |
|
{ |
|
public: |
|
Feature(); |
|
Feature( int offset, int x, int y, int cellW, int cellH ); |
|
float calc( const std::vector<cv::Mat> &_hists, const cv::Mat &_normSum, size_t y, int featComponent ) const; |
|
void write( cv::FileStorage &fs ) const; |
|
void write( cv::FileStorage &fs, int varIdx ) const; |
|
|
|
cv::Rect rect[N_CELLS]; //cells |
|
|
|
struct |
|
{ |
|
int p0, p1, p2, p3; |
|
} fastRect[N_CELLS]; |
|
}; |
|
std::vector<Feature> features; |
|
|
|
cv::Mat normSum; //for nomalization calculation (L1 or L2) |
|
std::vector<cv::Mat> hist; |
|
}; |
|
|
|
inline float CvHOGEvaluator::operator()(int varIdx, int sampleIdx) const |
|
{ |
|
int featureIdx = varIdx / (N_BINS * N_CELLS); |
|
int componentIdx = varIdx % (N_BINS * N_CELLS); |
|
//return features[featureIdx].calc( hist, sampleIdx, componentIdx); |
|
return features[featureIdx].calc( hist, normSum, sampleIdx, componentIdx); |
|
} |
|
|
|
inline float CvHOGEvaluator::Feature::calc( const std::vector<cv::Mat>& _hists, const cv::Mat& _normSum, size_t y, int featComponent ) const |
|
{ |
|
float normFactor; |
|
float res; |
|
|
|
int binIdx = featComponent % N_BINS; |
|
int cellIdx = featComponent / N_BINS; |
|
|
|
const float *phist = _hists[binIdx].ptr<float>((int)y); |
|
res = phist[fastRect[cellIdx].p0] - phist[fastRect[cellIdx].p1] - phist[fastRect[cellIdx].p2] + phist[fastRect[cellIdx].p3]; |
|
|
|
const float *pnormSum = _normSum.ptr<float>((int)y); |
|
normFactor = (float)(pnormSum[fastRect[0].p0] - pnormSum[fastRect[1].p1] - pnormSum[fastRect[2].p2] + pnormSum[fastRect[3].p3]); |
|
res = (res > 0.001f) ? ( res / (normFactor + 0.001f) ) : 0.f; //for cutting negative values, which apper due to floating precision |
|
|
|
return res; |
|
} |
|
|
|
#endif // _OPENCV_HOGFEATURES_H_
|
|
|