Mercurial Hosting > traffic-intelligence
diff python/cvutils.py @ 398:3399bd48cb40
Ajout d'une méthode pour obtenir le nombre de FPS
Méthode de capture des trames vidéos plus résistante aux erreur
Utilisation d'un dictionnaire pour les fichier de configuration afin de garder le nom des sections
| author | Jean-Philippe Jodoin <jpjodoin@gmail.com> |
|---|---|
| date | Mon, 29 Jul 2013 13:46:07 -0400 |
| parents | 167f6ec44ec5 |
| children | f29204e68aab |
line wrap: on
line diff
--- a/python/cvutils.py Sun Jul 28 18:04:20 2013 -0400 +++ b/python/cvutils.py Mon Jul 29 13:46:07 2013 -0400 @@ -140,8 +140,8 @@ from math import floor, log10 images = [] capture = cv2.VideoCapture(videoFilename) - nDigits = int(floor(log10(capture.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT))))+1 - if capture.isOpened(): + if capture.isOpened(): + nDigits = int(floor(log10(capture.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT))))+1 ret = False capture.set(cv2.cv.CV_CAP_PROP_POS_FRAMES, firstFrameNum) imgNum = 0 @@ -158,8 +158,22 @@ else: images.append(img) imgNum +=1 + capture.release() + else: + print('Video capture failed') return images + + def getFPS(videoFilename): + capture = cv2.VideoCapture(videoFilename) + if capture.isOpened(): + fps = capture.get(cv2.cv.CV_CAP_PROP_FPS) + capture.release() + return fps + else: + print 'Cannot load file ' + videoFilename + return None + def displayTrajectories(videoFilename, objects, boundingBoxes, homography = None, firstFrameNum = 0, lastFrameNumArg = None, printFrames = True, rescale = 1.): '''Displays the objects overlaid frame by frame over the video ''' capture = cv2.VideoCapture(videoFilename)