imgproc: local "CV_Assert(totalSampleCount > 0)" check

6 years ago · 5ed7d5a5d9
parent 1f7728db35
commit 5ed7d5a5d9
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;