integrate pruning

This commit is contained in:
marina.kolpakova
2012-12-11 22:42:13 +04:00
parent a89299acb2
commit 2e4b8d07cc
3 changed files with 113 additions and 15 deletions

View File

@@ -130,10 +130,15 @@ int main(int argc, char** argv)
if (boost.train(dataset, pool, cfg.weaks, cfg.treeDepth))
{
std::cout << "Octave " << *it << " was successfully trained..." << std::endl;
CvFileStorage* fout = cvOpenFileStorage(cfg.resPath(it).c_str(), 0, CV_STORAGE_WRITE);
boost.write(fout, cfg.cascadeName);
CvFileStorage* fout = cvOpenFileStorage(cfg.resPath(it).c_str(), 0, CV_STORAGE_WRITE);
boost.write(fout, cfg.cascadeName);
// strong.push_back(octave);
cvReleaseFileStorage( &fout);
cvReleaseFileStorage( &fout);
cv::Mat thresholds;
boost.setRejectThresholds(thresholds);
std::cout << "thresholds " << thresholds << std::endl;
}
}