Mercurial Hosting > traffic-intelligence
diff scripts/play-video.py @ 385:1917db662aa7
added rescaling options to scripts play-video and display-trajectories
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Mon, 22 Jul 2013 18:33:47 -0400 |
parents | 1d90e9080cb2 |
children | 3058e00887bc |
line wrap: on
line diff
--- a/scripts/play-video.py Mon Jul 22 18:11:01 2013 -0400 +++ b/scripts/play-video.py Mon Jul 22 18:33:47 2013 -0400 @@ -1,22 +1,23 @@ #! /usr/bin/env python -import sys,getopt +import sys, argparse import cvutils -options, args = getopt.getopt(sys.argv[1:], 'hi:f:',['help', 'fps=']) -options = dict(options) -print options -if '--help' in options.keys() or '-h' in options.keys() or len(sys.argv) == 1: - print('Usage: '+sys.argv[0]+' --help|-h -i video-filename [-f first_frame] [--fps frame_rate]') - sys.exit() +parser = argparse.ArgumentParser(description='The program displays the video.') +parser.add_argument('-i', dest = 'videoFilename', help = 'name of the video file', required = True) +parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', default = 0, type = int) +parser.add_argument('--fps', dest = 'frameRate', help = 'approximate frame rate to replay', type = float) +parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float) + +args = parser.parse_args() firstFrameNum = 0 -if '-f' in options.keys(): - firstFrameNum = int(options['-f']) +if args.firstFrameNum != None: + firstFrameNum = args.firstFrameNum frameRate = -1 -if '--fps' in options.keys(): - frameRate = int(options['--fps']) +if args.frameRate != None: + frameRate = args.frameRate -cvutils.playVideo(options['-i'], firstFrameNum, frameRate) +cvutils.playVideo(args.videoFilename, firstFrameNum, frameRate, rescale = args.rescale)