Commit b497380a authored by Andy Maloney's avatar Andy Maloney

Check memory allocation

Initialize local variables
parent 2a669555
...@@ -100,9 +100,9 @@ bool CvKNearest::train( const CvMat* _train_data, const CvMat* _responses, ...@@ -100,9 +100,9 @@ bool CvKNearest::train( const CvMat* _train_data, const CvMat* _responses,
__BEGIN__; __BEGIN__;
CvVectors* _samples; CvVectors* _samples = 0;
float** _data; float** _data = 0;
int _count, _dims, _dims_all, _rsize; int _count = 0, _dims = 0, _dims_all = 0, _rsize = 0;
if( !_update_base ) if( !_update_base )
clear(); clear();
...@@ -114,6 +114,9 @@ bool CvKNearest::train( const CvMat* _train_data, const CvMat* _responses, ...@@ -114,6 +114,9 @@ bool CvKNearest::train( const CvMat* _train_data, const CvMat* _responses,
_responses, CV_VAR_ORDERED, 0, _sample_idx, true, (const float***)&_data, _responses, CV_VAR_ORDERED, 0, _sample_idx, true, (const float***)&_data,
&_count, &_dims, &_dims_all, &responses, 0, 0 )); &_count, &_dims, &_dims_all, &responses, 0, 0 ));
if( !responses )
CV_ERROR( CV_StsNoMem, "Could not allocate memory for responses" );
if( _update_base && _dims != var_count ) if( _update_base && _dims != var_count )
CV_ERROR( CV_StsBadArg, "The newly added data have different dimensionality" ); CV_ERROR( CV_StsBadArg, "The newly added data have different dimensionality" );
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment