Mercurial Hosting > traffic-intelligence
changeset 864:8f28b4fcf129
minor
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 09 Nov 2016 18:00:39 -0500 |
parents | a8ca72dc1564 |
children | 5afa1d30edd8 |
files | python/cvutils.py |
diffstat | 1 files changed, 5 insertions(+), 2 deletions(-) [+] |
line wrap: on
line diff
--- a/python/cvutils.py Tue Nov 08 17:59:40 2016 -0500 +++ b/python/cvutils.py Wed Nov 09 18:00:39 2016 -0500 @@ -302,7 +302,7 @@ return croppedImg, yCropMin, yCropMax, xCropMin, xCropMax - def displayTrajectories(videoFilename, objects, boundingBoxes = {}, homography = None, firstFrameNum = 0, lastFrameNumArg = None, printFrames = True, rescale = 1., nFramesStep = 1, saveAllImages = False, undistort = False, intrinsicCameraMatrix = None, distortionCoefficients = None, undistortedImageMultiplication = 1., annotations = [], gtMatches = {}, toMatches = {}): + def displayTrajectories(videoFilename, objects, boundingBoxes = {}, homography = None, firstFrameNum = 0, lastFrameNumArg = None, printFrames = True, rescale = 1., nFramesStep = 1, saveAllImages = False, nZerosFilenameArg = None, undistort = False, intrinsicCameraMatrix = None, distortionCoefficients = None, undistortedImageMultiplication = 1., annotations = [], gtMatches = {}, toMatches = {}): '''Displays the objects overlaid frame by frame over the video ''' capture = cv2.VideoCapture(videoFilename) width = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH)) @@ -323,7 +323,10 @@ lastFrameNum = maxint else: lastFrameNum = lastFrameNumArg - nZerosFilename = int(ceil(log10(lastFrameNum))) + if nZerosFilenameArg is None: + nZerosFilename = int(ceil(log10(lastFrameNum))) + else: + nZerosFilename = nZerosFilenameArg objectToDeleteIds = [] while ret and not quitKey(key) and frameNum <= lastFrameNum: ret, img = capture.read()