cvpr读书笔记[3]:traincascade与AdaBoost的opencv实现框架

本节研究traincascade的opencv实现.

涉及的源代码位于:

sources\apps\traincascade traincascade实现

sources\modules\ml opencv machine learning部分

sources\data\vec_files\trainingfaces_24-24.vec 正样本

http://blog.csdn.net/njzhujinhua/article/details/38377191

【1】Cascade框架

首先从main入手

[cpp]  view plain  copy

 

1. int main( int argc, char* argv[] )  
2. {  
3.     CvCascadeClassifier classifier;  
4.     string cascadeDirName, vecName, bgName;  
5.     int numPos = 2000;  
6.     int numNeg = 1000;  
7.     int numStages = 20;  
8.     int precalcValBufSize = 256,  
9.         precalcIdxBufSize = 256;  
10.     bool baseFormatSave = false;  
11.   
12.     CvCascadeParams cascadeParams;  
13.     CvCascadeBoostParams stageParams;  
14.     Ptr<CvFeatureParams> featureParams[] = { Ptr<CvFeatureParams>(new CvHaarFeatureParams),  
15.                                              Ptr<CvFeatureParams>(new CvLBPFeatureParams),  
16.                                              Ptr<CvFeatureParams>(new CvHOGFeatureParams)  
17.                                            };  
18.     int fc = sizeof(featureParams)/sizeof(featureParams[0]);  
19. //略  
20.     classifier.train( cascadeDirName,  
21.                       vecName,  
22.                       bgName,  
23.                       numPos, numNeg,  
24.                       precalcValBufSize, precalcIdxBufSize,  
25.                       numStages,  
26.                       cascadeParams,  
27.                       *featureParams[cascadeParams.featureType],  
28.                       stageParams,  
29.                       baseFormatSave );  
30.     return 0;  
31. }  

其中的CvCascadeParams 继承自ml的CvParams,

CvCascadeBoostParams 继承自ml的CvBoostParams后者则是CvDTreeParams的子类.

 在CvCascadeBoostParams 中将boosttype设为了GENTLE,一类比较高效的AdaBoost

train的各参数:

cascadeDirName, 表示训练结果输出目录

vecName, 正样本的vec文件,由 opencv_createsamples 生成。正样本可以由包含待检测物体的一张图片生成，也可由一系列标记好的图像生成。

bgName, 背景图像的描述文件，文件中包含一系列的图像文件名，这些图像将被随机选作物体的背景

numPos, numNeg, 正负样本的个数

precalcValBufSize, 缓存大小，用于存储预先计算的特征值(feature values)，单位为MB。

precalcIdxBufSize 缓存大小，用于存储预先计算的特征索引(feature indices)，单位为MB。内存越大，训练时间越短。

numStages, 训练的分类器的级数

cascadeParams, 级联参数,除了默认值外,还可以通过参数指定. 其中stageType智能取值BOOST, featureType则支持haar,LBP,LOG

*featureParams[cascadeParams.featureType], 根据fratureType确定具体使用的FeatureParams

stageParams, boost分类器的参数,

     -bt指定boosttype,取值
         DAB=Discrete AdaBoost
         RAB = Real AdaBoost,
         LB = LogitBoost,
         GAB = Gentle AdaBoost,默认为GENTLE AdaBoost
     -minHitRate
     分类器的每一级最小检测率, 默认0.995。总的检测率大约为 min_hit_rate^number_of_stages。
     -maxFalseAlarmRate
     分类器的每一级允许最大FPR,默认0.5。总的为 max_false_alarm_rate^number_of_stages.
     -weightTrimRate
     样本权重按大小序累计超过此致的样本保留进入下一轮训练. 默认0.95。 见CvBoost::trim_weights
     -maxDepth
     弱分类器树最大的深度。默认是1，是二叉树（stumps）,只使用一个特征。
     -maxWeakCount
     每一级中的弱分类器的最大数目。默认100

baseFormatSave 这个参数仅在使用Haar特征时有效。如果指定这个参数，那么级联分类器将以老的格式存储。

CvCascadeClassifier::train()概述了整个Cascade的执行过程。包括训练前的初始化，各Stage的强分类器间的样本集更新及强分类器训练都可看到其踪影，最显眼的还是其中的Stage训练的for大循环。

[cpp]  view plain  copy

 

1. bool CvCascadeClassifier::train(...)  
2. {  
3. ...  
4. 读取正负样本  
5.     if ( !imgReader.create( _posFilename, _negFilename, _cascadeParams.winSize ) )  
6.     {  
7.         cout << "Image reader can not be created from -vec " << _posFilename<< " and -bg " << _negFilename << "." << endl;  
8.         return false;  
9.     }  
10. 在指定data目录中读取已训练过的stagexml文件  
11.     if ( !load( dirName ) )  
12.     {  
13. 文件不存在则执行初始化过程,  
14.         cascadeParams = _cascadeParams;  
15. 具体特征类型的创建与初始化  
16.         featureParams = CvFeatureParams::create(cascadeParams.featureType);  
17.         featureParams->init(_featureParams);  
18.         stageParams = new CvCascadeBoostParams;  
19.         *stageParams = _stageParams; 还是采用main函数里面获取的命令行个参数取值  
20. 特征计算器  
21.         featureEvaluator = CvFeatureEvaluator::create(cascadeParams.featureType);  
22. 特征的初始化,里面生成了24*24正样本区域的162336个特征.  
23.         featureEvaluator->init( (CvFeatureParams*)featureParams, numPos + numNeg, cascadeParams.winSize );  
24.         stageClassifiers.reserve( numStages );  
25.     }  
26. ......  
27. 打印参数  
28. <span style="color:#339999;">PARAMETERS:  
29. cascadeDirName: .  
30. vecFileName: trainingfaces_24-24.vec  
31. bgFileName: bg.txt  
32. numPos: 100  
33. numNeg: 120  
34. numStages: 20  
35. precalcValBufSize[Mb] : 256  
36. precalcIdxBufSize[Mb] : 256  
37. stageType: BOOST  
38. featureType: HAAR 特征类型  
39. sampleWidth: 24  
40. sampleHeight: 24  
41. boostType: GAB GENTLE AdaBoost  
42. minHitRate: 0.995  
43. maxFalseAlarmRate: 0.5  
44. weightTrimRate: 0.95  
45. maxDepth: 1 只有一个分支节点的二叉决策树  
46. maxWeakCount: 100  
47. mode: BASIC</span>  
48.   
49. 已经从文件读取的训练过的stage  
50.     int startNumStages = (int)stageClassifiers.size();  
51.     if ( startNumStages > 1 )  
52.         cout << endl << "Stages 0-" << startNumStages-1 << " are loaded" << endl;  
53.     else if ( startNumStages == 1)  
54.         cout << endl << "Stage 0 is loaded" << endl;  
55.   
56.     double requiredLeafFARate = pow( (double) stageParams->maxFalseAlarm, (double) numStages ) /  
57.                                 (double)stageParams->max_depth;  
58. 最终的虚警率  
59.     double tempLeafFARate;  
60.   
61.     for( int i = startNumStages; i < numStages; i++ )  
62.     {  
63.         cout << endl << "===== TRAINING " << i << "-stage =====" << endl;  
64.         cout << "<BEGIN" << endl;  
65.   
66.          更新stage间的样本, 要做到训练新的stage的时候仍保障正负样本数足够.  
67.         if ( !updateTrainingSet( tempLeafFARate ) )  
68.         {  
69.             样本数不够, 退出训练  
70.             cout << "Train dataset for temp stage can not be filled. Branch training terminated." << endl;  
71.             break;  
72.         }  
73.         if( tempLeafFARate <= requiredLeafFARate )  
74.         {  
75.               虚警率已经达标 不再继续训练  
76.             cout << "Required leaf false alarm rate achieved. Branch training terminated." << endl;  
77.             break;  
78.         }  
79.   
80.         CvCascadeBoost* tempStage = new CvCascadeBoost;  
81. cascade中一级的强分类器的训练过程  
82.         bool isStageTrained = tempStage->train( (CvFeatureEvaluator*)featureEvaluator,  
83.                                                 curNumSamples, _precalcValBufSize, _precalcIdxBufSize,  
84.                                                 *((CvCascadeBoostParams*)stageParams) );  
85.         cout << "END>" << endl;  
86.   
87.         if(!isStageTrained)  
88.             break;  
89.   
90.         stageClassifiers.push_back( tempStage );  
91.   
92.         //保存阶段性的stage到独立xml文件  
93.        ............  
94.     }  
95.   
96.     if(stageClassifiers.size() == 0)  
97.     {  
98.         cout << "Cascade classifier can't be trained. Check the used training parameters." << endl;  
99.         return false;  
100.     }  
101.    生成最终的xml文件  
102.     save( dirName + CC_CASCADE_FILENAME, baseFormatSave );  
103.   
104.     return true;  
105. }  

在介绍AdaBoost的实现前本应该先介绍下CvCascadeClassifier::updateTrainingSet的， 但因为其中的fillPassedSamples用到了CvCascadeClassifier::predict, 所以这个我们还是后面再讲吧。

【2】AdaBoost的训练过程

       AdaBoost由CvCascadeBoost类实现， 其继承自ml的CvBoost。

调用位置在CvCascadeClassifier::train中训练各Stage时调用

[cpp]  view plain  copy

 

1. CvCascadeBoost* tempStage = new CvCascadeBoost;  
2. bool isStageTrained = tempStage->train( (CvFeatureEvaluator*)featureEvaluator,  
3.                                         curNumSamples, _precalcValBufSize, _precalcIdxBufSize,  
4.                                         *((CvCascadeBoostParams*)stageParams) );  

 参数含义

 (CvFeatureEvaluator*)featureEvaluator, 特征提取器实现， HAAR LBP HOG

 curNumSamples, 正负样本总数

 _precalcValBufSize, _precalcIdxBufSize,前有说明，略

 *((CvCascadeBoostParams*)stageParams) Boost实现的一些参数， 包括AdaBoost的类型GENTLE等， 前面已有说明了

 

 

        在设计实现AdaBoost时最常用的弱分类器是决策树, 一般使用只含有一个节点的决策树就足够了,这就是所谓的stump(树桩).

 算法描述中得到的强分类器有一系列改变权重的样本训练得到的弱分类器按一定系数的线性组合，涉及到我们现在分析的CvCascadeBoost实现便是由一系列CvCascadeBoostTree表示的弱分类器。

 

[cpp]  view plain  copy

 

1. bool CvCascadeBoost::train(...)  
2. {  
3. ...  
4.     初始化Cascade中该级AdaBoost生成用到的训练数据。 此段比较耗时，后面分析  
5.     data = new CvCascadeBoostTrainData(...);  
6.   
7.     weak = cvCreateSeq( 0, sizeof(CvSeq), sizeof(CvBoostTree*), storage ); //CvSeq* weak 多个弱分类器CvCascadeBoostTree的序列  
8.   
9.  初始化第一个弱分类器前的权值  
10.     update_weights( 0 );  
11.   
12.     do  
13.     {  
14.      一个弱分类器及其训练过程  
15.         CvCascadeBoostTree* tree = new CvCascadeBoostTree;  
16.         tree->train( data, subsample_mask, this )  
17.         将训练好的弱分类器加入Seq中  
18.         cvSeqPush( weak, &tree );  
19.        更新样本权值  
20.         update_weights( tree );  
21.         trim_weights();  
22.         if( cvCountNonZero(subsample_mask) == 0 )  
23.             break;  
24.     }  
25.     while( !isErrDesired() && (weak->total < params.weak_count) );  
26.   
27.     if(weak->total > 0)  
28.     {  
29.         data->is_classifier = true;  
30.         data->free_train_data();  
31.         isTrained = true;  
32.     }  
33.     else  
34.         clear();  
35.   
36.     return isTrained;  
37. }  

其中的update_weights最开始参数为0时表示初始化权值为1.0/n

在训练完一个弱分类器后的update_weights(tree)则表示对权值的更新

void CvCascadeBoost::update_weights( CvBoostTree* tree )在权值更新流程时主要操作如下

[cpp]  view plain  copy

 

1. <pre name="code" class="cpp">for( int i = 0; i < n; i++ )  
2.     weak_eval->data.db[i] *= -orig_response->data.i[i];  
3.   
4. cvExp( weak_eval, weak_eval );  
5.   
6. for( int i = 0; i < n; i++ )  
7. {  
8.     double w = weights->data.db[i] * weak_eval->data.db[i];  
9.     weights->data.db[i] = w;  
10.     sumW += w;  
11. }  
12.   
13.    
14. // renormalize weights   权值归一化  
15. if( sumW > FLT_EPSILON )  
16. {  
17.     sumW = 1./sumW;  
18.     for( int i = 0; i < n; ++i )  
19.         weights->data.db[i] *= sumW;  
20. }  

其中weak_eval表示弱分类器的系数, 但看实现是回归树,并不同于公式中的\alpha_k=0.5*ln((1-Ek)/Ek). 具体取值的计算还没搞明白.

上述代码第二个for循环及其之前描述的是newweight[i]=oldweight[i]*exp(-weak_eval[i]*orig_response[i])

最后的if是权值归一化过程,还是与文档描述很一致的.