Merge pull request #13323 from alalek:issue_13297

6 years ago · ccf96b9e05
parent 3fe70d5cd5 5ed7d5a5d9
commit ccf96b9e05
1 changed files with 1 additions and 1 deletions
--- a/modules/imgproc/src/grabcut.cpp
+++ b/modules/imgproc/src/grabcut.cpp
@ -174,7 +174,6 @@ void GMM::addSample( int ci, const Vec3d color )

 void GMM::endLearning()
 {
-    CV_Assert(totalSampleCount > 0);
    for( int ci = 0; ci < componentsCount; ci++ )
    {
        int n = sampleCounts[ci];
@ -182,6 +181,7 @@ void GMM::endLearning()
            coefs[ci] = 0;
        else
        {
+            CV_Assert(totalSampleCount > 0);
            double inv_n = 1.0 / n;
            coefs[ci] = (double)n/totalSampleCount;