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

diff --git a/modules/imgproc/src/grabcut.cpp b/modules/imgproc/src/grabcut.cpp
index 9cfbe0d..d68077d 100644 (file)
--- 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;