Fix a memory leak in CvNormalBayesClassifier::train()

6fb3b4f4 · Csaba Kertesz · d08b1636 · 6fb3b4f4
Commit 6fb3b4f4 authored Sep 18, 2013 by Csaba Kertesz
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

nbayes.cpp modules/ml/src/nbayes.cpp +2 -0

No files found.
--- a/modules/ml/src/nbayes.cpp
+++ b/modules/ml/src/nbayes.cpp
@@ -210,6 +210,8 @@ bool CvNormalBayesClassifier::train( const CvMat* _train_data, const CvMat* _res
                prod_data[c2] += train_vec[c2]*val1;
        }
    }
+    cvReleaseMat( &responses );
+    responses = 0;

    /* calculate avg, covariance matrix, c */
    for( cls = 0; cls < nclasses; cls++ )