apps/traincascade/lbpfeatures.h

   1 #ifndef _OPENCV_LBPFEATURES_H_
   2 #define _OPENCV_LBPFEATURES_H_
   3
   4 #include "traincascade_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, cv::Size _winSize );
  19     virtual void setImage(const cv::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( cv::FileStorage &fs, const cv::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 cv::Mat& _sum, size_t y ) const;
  32         void write( cv::FileStorage &fs ) const;
  33
  34         cv::Rect rect;
  35         int p[16];
  36     };
  37     std::vector<Feature> features;
  38
  39     cv::Mat sum;
  40 };
  41
  42 inline uchar CvLBPEvaluator::Feature::calc(const cv::Mat &_sum, size_t y) const
  43 {
  44     const int* psum = _sum.ptr<int>((int)y);
  45     int cval = psum[p[5]] - psum[p[6]] - psum[p[9]] + psum[p[10]];
  46
  47     return (uchar)((psum[p[0]] - psum[p[1]] - psum[p[4]] + psum[p[5]] >= cval ? 128 : 0) |   // 0
  48         (psum[p[1]] - psum[p[2]] - psum[p[5]] + psum[p[6]] >= cval ? 64 : 0) |    // 1
  49         (psum[p[2]] - psum[p[3]] - psum[p[6]] + psum[p[7]] >= cval ? 32 : 0) |    // 2
  50         (psum[p[6]] - psum[p[7]] - psum[p[10]] + psum[p[11]] >= cval ? 16 : 0) |  // 5
  51         (psum[p[10]] - psum[p[11]] - psum[p[14]] + psum[p[15]] >= cval ? 8 : 0) | // 8
  52         (psum[p[9]] - psum[p[10]] - psum[p[13]] + psum[p[14]] >= cval ? 4 : 0) |  // 7
  53         (psum[p[8]] - psum[p[9]] - psum[p[12]] + psum[p[13]] >= cval ? 2 : 0) |   // 6
  54         (psum[p[4]] - psum[p[5]] - psum[p[8]] + psum[p[9]] >= cval ? 1 : 0));     // 3
  55 }
  56
  57 #endif