mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 06:03:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #420 from asmaloney:check-mem-alloc

Browse Source
This commit is contained in:
Andrey Kamaev 2013-02-04 12:56:35 +04:00 committed by OpenCV Buildbot
commit 9c939a8dcf
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/ml/src/knearest.cpp
View File

@ -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,
_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" );