Mercurial Hosting > traffic-intelligence

changeset 799:0662c87a61c9 dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
minor modification
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 30 May 2016 23:01:53 -0400
parents 5b99b676265e
children 2cade72d75ad
files python/cvutils.py
diffstat 1 files changed, 2 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/python/cvutils.py	Mon May 30 22:49:24 2016 -0400
+++ b/python/cvutils.py	Mon May 30 23:01:53 2016 -0400
@@ -230,13 +230,13 @@
             ret = False
             capture.set(cv2.CAP_PROP_POS_FRAMES, firstFrameNum)
             frameNum = firstFrameNum
-            while frameNum<=lastFrameNum:
+            while frameNum<=lastFrameNum and frameNum<rawCount:
                 ret, img = capture.read()
                 i = 0
                 while not ret and i<10:
                     ret, img = capture.read()
                     i += 1
-                if img.size>0:
+                if img is not None and img.size>0:
                     if saveImage:
                         frameNumStr = format(frameNum, '0{}d'.format(nDigits))
                         cv2.imwrite(outputPrefix+frameNumStr+'.png', img)