Merge pull request #23019 from tkram01:sampleIdxFix

Fix for using sampleIdx to limit training data
2 years ago · 8c64adb000
parent a94cd6d6e8 ea7efd57d8
commit 8c64adb000
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 )