opencv实例训练器traincascade源码学习-------traincascade.cpp解析
1 #include "opencv2/core.hpp" 2 #include "cascadeclassifier.h" 3 4 using namespace std; 5 using namespace cv; 6 7 int main( int argc, char* argv[] ) 8 { 9 //默认初始化 10 CvCascadeClassifier classifier; 11 string cascadeDirName, vecName, bgName; 12 int numPos = 2000; 13 int numNeg = 1000; 14 int numStages = 20; 15 int numThreads = getNumThreads(); 16 int precalcValBufSize = 1024, 17 precalcIdxBufSize = 1024; 18 bool baseFormatSave = false; 19 double acceptanceRatioBreakValue = -1.0; 20 21 CvCascadeParams cascadeParams; 22 CvCascadeBoostParams stageParams; 23 //三种分类方式的分类器指针 24 Ptr<CvFeatureParams> featureParams[] = { makePtr<CvHaarFeatureParams>(),//Harr 25 makePtr<CvLBPFeatureParams>(),//LBP 26 makePtr<CvHOGFeatureParams>()//HOG 27 }; 28 int fc = sizeof(featureParams)/sizeof(featureParams[0]);//计算分类器个数 29 //输出默认参数 30 if( argc == 1 ) 31 { 32 cout << "Usage: " << argv[0] << endl;//获取执行的.exe文件 33 cout << " -data <cascade_dir_name>" << endl; 34 cout << " -vec <vec_file_name>" << endl; 35 cout << " -bg <background_file_name>" << endl; 36 cout << " [-numPos <number_of_positive_samples = " << numPos << ">]" << endl; 37 cout << " [-numNeg <number_of_negative_samples = " << numNeg << ">]" << endl; 38 cout << " [-numStages <number_of_stages = " << numStages << ">]" << endl; 39 cout << " [-precalcValBufSize <precalculated_vals_buffer_size_in_Mb = " << precalcValBufSize << ">]" << endl; 40 cout << " [-precalcIdxBufSize <precalculated_idxs_buffer_size_in_Mb = " << precalcIdxBufSize << ">]" << endl; 41 cout << " [-baseFormatSave]" << endl; 42 cout << " [-numThreads <max_number_of_threads = " << numThreads << ">]" << endl; 43 cout << " [-acceptanceRatioBreakValue <value> = " << acceptanceRatioBreakValue << ">]" << endl; 44 cascadeParams.printDefaults(); 45 stageParams.printDefaults(); 46 for( int fi = 0; fi < fc; fi++ ) 47 featureParams[fi]->printDefaults();//打印各个分类器的默认参数 48 return 0; 49 } 50 //命令解析 51 for( int i = 1; i < argc; i++ ) 52 { 53 bool set = false; 54 if( !strcmp( argv[i], "-data" ) ) 55 { 56 cascadeDirName = argv[++i]; 57 } 58 else if( !strcmp( argv[i], "-vec" ) ) 59 { 60 vecName = argv[++i]; 61 } 62 else if( !strcmp( argv[i], "-bg" ) ) 63 { 64 bgName = argv[++i]; 65 } 66 else if( !strcmp( argv[i], "-numPos" ) ) 67 { 68 numPos = atoi( argv[++i] ); 69 } 70 else if( !strcmp( argv[i], "-numNeg" ) ) 71 { 72 numNeg = atoi( argv[++i] ); 73 } 74 else if( !strcmp( argv[i], "-numStages" ) ) 75 { 76 numStages = atoi( argv[++i] ); 77 } 78 else if( !strcmp( argv[i], "-precalcValBufSize" ) ) 79 { 80 precalcValBufSize = atoi( argv[++i] ); 81 } 82 else if( !strcmp( argv[i], "-precalcIdxBufSize" ) ) 83 { 84 precalcIdxBufSize = atoi( argv[++i] ); 85 } 86 else if( !strcmp( argv[i], "-baseFormatSave" ) ) 87 { 88 baseFormatSave = true; 89 } 90 else if( !strcmp( argv[i], "-numThreads" ) ) 91 { 92 numThreads = atoi(argv[++i]); 93 } 94 else if( !strcmp( argv[i], "-acceptanceRatioBreakValue" ) ) 95 { 96 acceptanceRatioBreakValue = atof(argv[++i]); 97 } 98 else if ( cascadeParams.scanAttr( argv[i], argv[i+1] ) ) { i++; } 99 else if ( stageParams.scanAttr( argv[i], argv[i+1] ) ) { i++; } 100 else if ( !set ) 101 { 102 for( int fi = 0; fi < fc; fi++ ) 103 { 104 set = featureParams[fi]->scanAttr(argv[i], argv[i+1]); 105 if ( !set ) 106 { 107 i++; 108 break; 109 } 110 } 111 } 112 } 113 114 setNumThreads( numThreads ); 115 classifier.train( cascadeDirName, 116 vecName, 117 bgName, 118 numPos, numNeg, 119 precalcValBufSize, precalcIdxBufSize, 120 numStages, 121 cascadeParams, 122 *featureParams[cascadeParams.featureType], 123 stageParams, 124 baseFormatSave, 125 acceptanceRatioBreakValue ); 126 return 0; 127 }
这部分主要是主函数对输入的命令参数的解析，及默认值输出。
主函数转向的线路图：
main（）-------->cascadeclassifier.h -------->调用这部分的函数 train（）进行数据训练。
初次尝试源码分析，多多指教!!
