fixed #1048

94a5bf88 · Maria Dimashova · 62b96646 · 94a5bf88
Commit 94a5bf88 authored May 06, 2011 by Maria Dimashova
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

grabcut.cpp modules/imgproc/src/grabcut.cpp +5 -1

No files found.
--- 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;
 }