Commit 46110c3d authored by Daniel Canelhas's avatar Daniel Canelhas Committed by Alexander Alekhin

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
parent 766bf772
...@@ -333,7 +333,14 @@ void WaldBoost::fit(Mat& data_pos, Mat& data_neg) ...@@ -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;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment