generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #1858 from StevenPuttemans:updates_traincascade

Browse Source
This commit is contained in:
Roman Donchenko 2013-11-27 17:37:36 +04:00 committed by OpenCV Buildbot
commit d5ead208b7
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
apps/traincascade/cascadeclassifier.cpp
View File

@ -137,6 +137,9 @@ bool CvCascadeClassifier::train( const String _cascadeDirName,
const CvCascadeBoostParams& _stageParams, const CvCascadeBoostParams& _stageParams,
bool baseFormatSave ) bool baseFormatSave )
{ {
// Start recording clock ticks for training time output
const clock_t begin_time = clock();
if( _cascadeDirName.empty() || _posFilename.empty() || _negFilename.empty() ) if( _cascadeDirName.empty() || _posFilename.empty() || _negFilename.empty() )
CV_Error( CV_StsBadArg, "_cascadeDirName or _bgfileName or _vecFileName is NULL" ); CV_Error( CV_StsBadArg, "_cascadeDirName or _bgfileName or _vecFileName is NULL" );
@ -247,6 +250,14 @@ bool CvCascadeClassifier::train( const String _cascadeDirName,
fs << FileStorage::getDefaultObjectName(stageFilename) << "{"; fs << FileStorage::getDefaultObjectName(stageFilename) << "{";
tempStage->write( fs, Mat() ); tempStage->write( fs, Mat() );
fs << "}"; fs << "}";
// Output training time up till now
float seconds = float( clock () - begin_time ) / CLOCKS_PER_SEC;
int days = int(seconds) / 60 / 60 / 24;
int hours = (int(seconds) / 60 / 60) % 24;
int minutes = (int(seconds) / 60) % 60;
int seconds_left = int(seconds) % 60;
cout << "Training until now has taken " << days << " days " << hours << " hours " << minutes << " minutes " << seconds_left <<" seconds." << endl;
} }
if(stageClassifiers.size() == 0) if(stageClassifiers.size() == 0)
@ -310,6 +321,7 @@ int CvCascadeClassifier::fillPassedSamples( int first, int count, bool isPositiv
if( predict( i ) == 1.0F ) if( predict( i ) == 1.0F )
{ {
getcount++; getcount++;
printf("%s current samples: %d\r", isPositive ? "POS":"NEG", getcount);
break; break;
} }
} }