Merge pull request #1993 from dcanelhas:master

waldboost_detector train: avoid failed asserts while training by exiting early

* Avoids a failed assert which crashes training

* xobjdetect: improve waldboost messages
pull/2282/head^2
Daniel Canelhas 6 years ago committed by Alexander Alekhin
parent 766bf7723a
commit 46110c3d9e
  1. 9
      modules/xobjdetect/src/waldboost.cpp

@ -333,7 +333,14 @@ void WaldBoost::fit(Mat& data_pos, Mat& data_neg)
if (loss < 1e-50 || min_err > 0.5) { if (loss < 1e-50 || min_err > 0.5) {
std::cerr << "Stopping early" << std::endl; std::cerr << "Stopping early. loss=" << loss << " min_err=" << min_err << std::endl;
weak_count_ = i + 1;
break;
}
// Avoid crashing on next Mat creation
if (pos <= 1) {
std::cerr << "Stopping early. pos=" << pos << std::endl;
weak_count_ = i + 1; weak_count_ = i + 1;
break; break;
} }

Loading…
Cancel
Save