1 #include "opencv2/core.hpp"
2 #include "opencv2/imgproc.hpp"
3
4 #include "lbpfeatures.h"
5 #include "cascadeclassifier.h"
6
7 using namespace cv;
8
CvLBPFeatureParams()9 CvLBPFeatureParams::CvLBPFeatureParams()
10 {
11 maxCatCount = 256;
12 name = LBPF_NAME;
13 }
14
init(const CvFeatureParams * _featureParams,int _maxSampleCount,Size _winSize)15 void CvLBPEvaluator::init(const CvFeatureParams *_featureParams, int _maxSampleCount, Size _winSize)
16 {
17 CV_Assert( _maxSampleCount > 0);
18 sum.create((int)_maxSampleCount, (_winSize.width + 1) * (_winSize.height + 1), CV_32SC1);
19 CvFeatureEvaluator::init( _featureParams, _maxSampleCount, _winSize );
20 }
21
setImage(const Mat & img,uchar clsLabel,int idx)22 void CvLBPEvaluator::setImage(const Mat &img, uchar clsLabel, int idx)
23 {
24 CV_DbgAssert( !sum.empty() );
25 CvFeatureEvaluator::setImage( img, clsLabel, idx );
26 Mat innSum(winSize.height + 1, winSize.width + 1, sum.type(), sum.ptr<int>((int)idx));
27 integral( img, innSum );
28 }
29
writeFeatures(FileStorage & fs,const Mat & featureMap) const30 void CvLBPEvaluator::writeFeatures( FileStorage &fs, const Mat& featureMap ) const
31 {
32 _writeFeatures( features, fs, featureMap );
33 }
34
generateFeatures()35 void CvLBPEvaluator::generateFeatures()
36 {
37 int offset = winSize.width + 1;
38 for( int x = 0; x < winSize.width; x++ )
39 for( int y = 0; y < winSize.height; y++ )
40 for( int w = 1; w <= winSize.width / 3; w++ )
41 for( int h = 1; h <= winSize.height / 3; h++ )
42 if ( (x+3*w <= winSize.width) && (y+3*h <= winSize.height) )
43 features.push_back( Feature(offset, x, y, w, h ) );
44 numFeatures = (int)features.size();
45 }
46
Feature()47 CvLBPEvaluator::Feature::Feature()
48 {
49 rect = cvRect(0, 0, 0, 0);
50 }
51
Feature(int offset,int x,int y,int _blockWidth,int _blockHeight)52 CvLBPEvaluator::Feature::Feature( int offset, int x, int y, int _blockWidth, int _blockHeight )
53 {
54 Rect tr = rect = cvRect(x, y, _blockWidth, _blockHeight);
55 CV_SUM_OFFSETS( p[0], p[1], p[4], p[5], tr, offset )
56 tr.x += 2*rect.width;
57 CV_SUM_OFFSETS( p[2], p[3], p[6], p[7], tr, offset )
58 tr.y +=2*rect.height;
59 CV_SUM_OFFSETS( p[10], p[11], p[14], p[15], tr, offset )
60 tr.x -= 2*rect.width;
61 CV_SUM_OFFSETS( p[8], p[9], p[12], p[13], tr, offset )
62 }
63
write(FileStorage & fs) const64 void CvLBPEvaluator::Feature::write(FileStorage &fs) const
65 {
66 fs << CC_RECT << "[:" << rect.x << rect.y << rect.width << rect.height << "]";
67 }
68