git.maemo.org Git - opencv/blob - apps/traincascade/lbpfeatures.h

   1 #ifndef _OPENCV_LBPFEATURES_H_
   2 #define _OPENCV_LBPFEATURES_H_
   3
   4 #include "features.h"
   5
   6 #define LBPF_NAME "lbpFeatureParams"
   7 struct CvLBPFeatureParams : CvFeatureParams
   8 {
   9     CvLBPFeatureParams();
  10
  11 };
  12
  13 class CvLBPEvaluator : public CvFeatureEvaluator
  14 {
  15 public:
  16     virtual ~CvLBPEvaluator() {}
  17     virtual void init(const CvFeatureParams *_featureParams,
  18         int _maxSampleCount, Size _winSize );
  19     virtual void setImage(const Mat& img, uchar clsLabel, int idx);
  20     virtual float operator()(int featureIdx, int sampleIdx) const
  21     { return (float)features[featureIdx].calc( sum, sampleIdx); }
  22     virtual void writeFeatures( FileStorage &fs, const Mat& featureMap ) const;
  23 protected:
  24     virtual void generateFeatures();
  25
  26     class Feature
  27     {
  28     public:
  29         Feature();
  30         Feature( int offset, int x, int y, int _block_w, int _block_h  );
  31         uchar calc( const Mat& _sum, size_t y ) const;
  32         void write( FileStorage &fs ) const;
  33
  34         Rect rect;
  35         int p[16];
  36     };
  37     vector<Feature> features;
  38
  39     Mat sum;
  40 };
  41
  42 inline uchar CvLBPEvaluator::Feature::calc(const Mat &_sum, size_t y) const
  43 {
  44     const int* sum = _sum.ptr<int>((int)y);
  45     int cval = sum[p[5]] - sum[p[6]] - sum[p[9]] + sum[p[10]];
  46
  47     return (uchar)((sum[p[0]] - sum[p[1]] - sum[p[4]] + sum[p[5]] >= cval ? 128 : 0) |   // 0
  48         (sum[p[1]] - sum[p[2]] - sum[p[5]] + sum[p[6]] >= cval ? 64 : 0) |    // 1
  49         (sum[p[2]] - sum[p[3]] - sum[p[6]] + sum[p[7]] >= cval ? 32 : 0) |    // 2
  50         (sum[p[6]] - sum[p[7]] - sum[p[10]] + sum[p[11]] >= cval ? 16 : 0) |  // 5
  51         (sum[p[10]] - sum[p[11]] - sum[p[14]] + sum[p[15]] >= cval ? 8 : 0) | // 8
  52         (sum[p[9]] - sum[p[10]] - sum[p[13]] + sum[p[14]] >= cval ? 4 : 0) |  // 7
  53         (sum[p[8]] - sum[p[9]] - sum[p[12]] + sum[p[13]] >= cval ? 2 : 0) |   // 6
  54         (sum[p[4]] - sum[p[5]] - sum[p[8]] + sum[p[9]] >= cval ? 1 : 0));     // 3
  55 }
  56
  57 #endif