fixed vector out of range exception in nextImg

chenbk85 · May 18, 2015 · 5cb70d3 · 5cb70d3
1 parent 354ed5c
commit 5cb70d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/apps/traincascade/imagestorage.cpp b/apps/traincascade/imagestorage.cpp
@@ -54,8 +54,10 @@ bool CvCascadeImageReader::NegReader::nextImg()
     for( size_t i = 0; i < count; i++ )
     {
         src = imread( imgFilenames[last++], 0 );
-        if( src.empty() )
+        if( src.empty() ){
+            last %= count;
             continue;
+        }
         round += last / count;
         round = round % (winSize.width * winSize.height);
         last %= count;