fixed CvCascadeBoost::train

44e05308 · Maria Dimashova · 1f3f4599 · 44e05308
Commit 44e05308 authored Nov 16, 2011 by Maria Dimashova
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

boost.cpp modules/traincascade/boost.cpp +2 -0

No files found.
--- a/modules/traincascade/boost.cpp
+++ b/modules/traincascade/boost.cpp
@@ -1016,6 +1016,8 @@ bool CvCascadeBoost::train( const CvFeatureEvaluator* _featureEvaluator,
        cvSeqPush( weak, &tree );
        update_weights( tree );
        trim_weights();
+        if( cvCountNonZero(subsample_mask) == 0 )
+            break;
    }
    while( !isErrDesired() && (weak->total < params.weak_count) );