Merge pull request #5151 from mshabunin:fix-training-data-corruption

pull/5193/head
Maksim Shabunin 9 years ago
commit 34aa4e4578
  1. 5
      modules/ml/src/inner_functions.cpp

@ -777,7 +777,8 @@ cvGetTrainSamples( const CvMat* train_data, int tflag,
__BEGIN__; __BEGIN__;
int i, j, var_count, sample_count, s_step, v_step; int i, j, var_count, sample_count;
size_t s_step, v_step, s;
bool copy_data; bool copy_data;
const float* data; const float* data;
const int *s_idx, *v_idx; const int *s_idx, *v_idx;
@ -815,7 +816,7 @@ cvGetTrainSamples( const CvMat* train_data, int tflag,
{ {
samples[0] = (float*)(samples + sample_count); samples[0] = (float*)(samples + sample_count);
if( tflag != CV_ROW_SAMPLE ) if( tflag != CV_ROW_SAMPLE )
CV_SWAP( s_step, v_step, i ); CV_SWAP( s_step, v_step, s );
for( i = 0; i < sample_count; i++ ) for( i = 0; i < sample_count; i++ )
{ {

Loading…
Cancel
Save