third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update bgfg_gaussmix2.cpp to avoid divide by zero cases.

Browse Source
This commit is contained in:
Matthias Grundmann 2017-03-08 19:27:43 -08:00 committed by GitHub
parent 8ef23d64a1
commit 0a1767a6b5
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
modules/video/src/bgfg_gaussmix2.cpp
View File

@ -693,11 +693,16 @@ public:
//go through all modes
//////
//renormalize weights
totalWeight = 1.f/totalWeight;
// Renormalize weights. In the special case that the pixel does
// not agree with any modes, set weights to zero (a new mode will be added below).
float invWeight = 0.f;
if (fabs(totalWeight) > 1e-12f) {
invWeight = 1.f/totalWeight;
}
for( int mode = 0; mode < nmodes; mode++ )
{
gmm[mode].weight *= totalWeight;
gmm[mode].weight *= invWeight;
}
//make new mode if needed and exit
@ -900,7 +905,10 @@ void BackgroundSubtractorMOG2Impl::getBackgroundImage_intern(OutputArray backgro
if(totalWeight > backgroundRatio)
break;
}
float invWeight = 1.f/totalWeight;
float invWeight = 0.f;
if (fabs(totalWeight) > 1e-12f) {
invWeight = 1.f/totalWeight;
}
meanBackground.at<Vec<T,CN> >(row, col) = Vec<T,CN>(meanVal * invWeight);
meanVal = 0.f;