Mercurial > hg > nsaunier > traffic-intelligence

annotate scripts/play-video.py @ 622:dc8490726d06

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected issues created with homography projection in Trajectory
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 22 Dec 2014 16:29:34 -0500
parents 1917db662aa7
children 3058e00887bc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
305
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
1 #! /usr/bin/env python
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
2
385
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
3 import sys, argparse
305
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
4 import cvutils
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
5
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
6
385
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
7 parser = argparse.ArgumentParser(description='The program displays the video.')
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
8 parser.add_argument('-i', dest = 'videoFilename', help = 'name of the video file', required = True)
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
9 parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', default = 0, type = int)
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
10 parser.add_argument('--fps', dest = 'frameRate', help = 'approximate frame rate to replay', type = float)
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
11 parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float)
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
12
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
13 args = parser.parse_args()
305
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
14
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
15 firstFrameNum = 0
385
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
16 if args.firstFrameNum != None:
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
17 firstFrameNum = args.firstFrameNum
305
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
18
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
19 frameRate = -1
385
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
20 if args.frameRate != None:
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
21 frameRate = args.frameRate
305
ca9131968bce added sample to replay video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
22
385
1917db662aa7 added rescaling options to scripts play-video and display-trajectories
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 334
diff changeset
23 cvutils.playVideo(args.videoFilename, firstFrameNum, frameRate, rescale = args.rescale)