Mercurial Hosting > traffic-intelligence

comparison python/cvutils.py @ 520:fd9641cbd24b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added function to classify object at instant from SVM
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 18 Jun 2014 01:19:18 -0400
parents 727e3c529519
children 5585ebd8ad61
comparison
equal deleted inserted replaced
519:4ad5123d969e 520:fd9641cbd24b
203 yCropMin = int(max(0, .5 * (ymin + ymax - a))) 203 yCropMin = int(max(0, .5 * (ymin + ymax - a)))
204 yCropMax = int(min(height - 1, .5 * (ymin + ymax + a))) 204 yCropMax = int(min(height - 1, .5 * (ymin + ymax + a)))
205 xCropMin = int(max(0, .5 * (xmin + xmax - a))) 205 xCropMin = int(max(0, .5 * (xmin + xmax - a)))
206 xCropMax = int(min(width - 1, .5 * (xmin + xmax + a))) 206 xCropMax = int(min(width - 1, .5 * (xmin + xmax + a)))
207 if yCropMax != yCropMin and xCropMax != xCropMin and (yCropMax - yCropMin) * (xCropMax - xCropMin) > pixelThreshold: 207 if yCropMax != yCropMin and xCropMax != xCropMin and (yCropMax - yCropMin) * (xCropMax - xCropMin) > pixelThreshold:
208 imgcrop = img[yCropMin : yCropMax, xCropMin : xCropMax] 208 croppedImg = img[yCropMin : yCropMax, xCropMin : xCropMax]
209 else: 209 else:
210 imgcrop = [] 210 croppedImg = []
211 return imgcrop, yCropMin, yCropMax, xCropMin, xCropMax 211 return imgcrop, yCropMin, yCropMax, xCropMin, xCropMax
212 212
213 213
214 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.): 214 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.):
215 '''Displays the objects overlaid frame by frame over the video ''' 215 '''Displays the objects overlaid frame by frame over the video '''