|
|
@ -174,7 +174,6 @@ void GMM::addSample( int ci, const Vec3d color ) |
|
|
|
|
|
|
|
|
|
|
|
void GMM::endLearning() |
|
|
|
void GMM::endLearning() |
|
|
|
{ |
|
|
|
{ |
|
|
|
CV_Assert(totalSampleCount > 0); |
|
|
|
|
|
|
|
for( int ci = 0; ci < componentsCount; ci++ ) |
|
|
|
for( int ci = 0; ci < componentsCount; ci++ ) |
|
|
|
{ |
|
|
|
{ |
|
|
|
int n = sampleCounts[ci]; |
|
|
|
int n = sampleCounts[ci]; |
|
|
@ -182,6 +181,7 @@ void GMM::endLearning() |
|
|
|
coefs[ci] = 0; |
|
|
|
coefs[ci] = 0; |
|
|
|
else |
|
|
|
else |
|
|
|
{ |
|
|
|
{ |
|
|
|
|
|
|
|
CV_Assert(totalSampleCount > 0); |
|
|
|
double inv_n = 1.0 / n; |
|
|
|
double inv_n = 1.0 / n; |
|
|
|
coefs[ci] = (double)n/totalSampleCount; |
|
|
|
coefs[ci] = (double)n/totalSampleCount; |
|
|
|
|
|
|
|
|
|
|
|