1 #ifndef _OPENCV_FEATURES_H_
2 #define _OPENCV_FEATURES_H_
4 #include "imagestorage.h"
10 #define FEATURES "features"
12 #define CV_SUM_OFFSETS( p0, p1, p2, p3, rect, step ) \
14 (p0) = (rect).x + (step) * (rect).y; \
16 (p1) = (rect).x + (rect).width + (step) * (rect).y; \
18 (p2) = (rect).x + (step) * ((rect).y + (rect).height); \
19 /* (x + w, y + h) */ \
20 (p3) = (rect).x + (rect).width + (step) * ((rect).y + (rect).height);
22 #define CV_TILTED_OFFSETS( p0, p1, p2, p3, rect, step ) \
24 (p0) = (rect).x + (step) * (rect).y; \
25 /* (x - h, y + h) */ \
26 (p1) = (rect).x - (rect).height + (step) * ((rect).y + (rect).height);\
27 /* (x + w, y + w) */ \
28 (p2) = (rect).x + (rect).width + (step) * ((rect).y + (rect).width); \
29 /* (x + w - h, y + w + h) */ \
30 (p3) = (rect).x + (rect).width - (rect).height \
31 + (step) * ((rect).y + (rect).width + (rect).height);
33 float calcNormFactor( const Mat& sum, const Mat& sqSum );
35 template<class Feature>
36 void _writeFeatures( const std::vector<Feature> features, FileStorage &fs, const Mat& featureMap )
38 fs << FEATURES << "[";
39 const Mat_<int>& featureMap_ = (const Mat_<int>&)featureMap;
40 for ( int fi = 0; fi < featureMap.cols; fi++ )
41 if ( featureMap_(0, fi) >= 0 )
44 features[fi].write( fs );
54 virtual ~CvParams() {}
56 virtual void write( FileStorage &fs ) const = 0;
57 virtual bool read( const FileNode &node ) = 0;
59 virtual void printDefaults() const;
60 virtual void printAttrs() const;
61 virtual bool scanAttr( const std::string prmName, const std::string val );
65 class CvFeatureParams : public CvParams
68 enum { HAAR = 0, LBP = 1, HOG = 2 };
70 virtual void init( const CvFeatureParams& fp );
71 virtual void write( FileStorage &fs ) const;
72 virtual bool read( const FileNode &node );
73 static Ptr<CvFeatureParams> create( int featureType );
74 int maxCatCount; // 0 in case of numerical features
75 int featSize; // 1 in case of simple features (HAAR, LBP) and N_BINS(9)*N_CELLS(4) in case of Dalal's HOG features
78 class CvFeatureEvaluator
81 virtual ~CvFeatureEvaluator() {}
82 virtual void init(const CvFeatureParams *_featureParams,
83 int _maxSampleCount, Size _winSize );
84 virtual void setImage(const Mat& img, uchar clsLabel, int idx);
85 virtual void writeFeatures( FileStorage &fs, const Mat& featureMap ) const = 0;
86 virtual float operator()(int featureIdx, int sampleIdx) const = 0;
87 static Ptr<CvFeatureEvaluator> create(int type);
89 int getNumFeatures() const { return numFeatures; }
90 int getMaxCatCount() const { return featureParams->maxCatCount; }
91 int getFeatureSize() const { return featureParams->featSize; }
92 const Mat& getCls() const { return cls; }
93 float getCls(int si) const { return cls.at<float>(si, 0); }
95 virtual void generateFeatures() = 0;
100 CvFeatureParams *featureParams;