fixed #1048

14 years ago · 94a5bf88d0
parent 62b966460d
commit 94a5bf88d0
1 changed files with 5 additions and 1 deletions
--- a/modules/imgproc/src/grabcut.cpp
+++ b/modules/imgproc/src/grabcut.cpp
@ -260,7 +260,11 @@ double calcBeta( const Mat& img )
            }
        }
    }
-    beta = 1.f / (2 * beta/(4*img.cols*img.rows - 3*img.cols - 3*img.rows + 2) );
+    if( beta <= std::numeric_limits<double>::epsilon() )
+        beta = 0;
+    else
+        beta = 1.f / (2 * beta/(4*img.cols*img.rows - 3*img.cols - 3*img.rows + 2) );
+
    return beta;
 }