Fix for using sampleIdx to limit training data

2 years ago · ea7efd57d8
parent a924bbfc30
commit ea7efd57d8
1 changed files with 1 additions and 1 deletions
--- a/modules/ml/src/data.cpp
+++ b/modules/ml/src/data.cpp
@ -904,7 +904,7 @@ public:
            if( s )
            {
                j = s[i];
-                CV_Assert( 0 <= j && j < nsamples );
+                CV_Assert( 0 <= j && j < ((layout == ROW_SAMPLE) ? samples.rows : samples.cols) );
            }
            values[i] = src[j*sstep];
            if( values[i] == MISSED_VAL )