Mercurial Hosting > traffic-intelligence

comparison scripts/display-trajectories.py @ 482:f6415f012640

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
adding functionalities (save images directly to display trajectories to create movies
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 02 Apr 2014 16:12:24 -0400
parents d337bffd7283
children 30fb60428e09
comparison
equal deleted inserted replaced
481:b6ad86ee7033 482:f6415f012640
14 parser.add_argument('-t', dest = 'trajectoryType', help = 'type of trajectories to display', choices = ['feature', 'object'], default = 'feature') 14 parser.add_argument('-t', dest = 'trajectoryType', help = 'type of trajectories to display', choices = ['feature', 'object'], default = 'feature')
15 parser.add_argument('-o', dest = 'homography', help = 'name of the image to world homography') 15 parser.add_argument('-o', dest = 'homography', help = 'name of the image to world homography')
16 parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', default = 0, type = int) 16 parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', default = 0, type = int)
17 parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float) 17 parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float)
18 parser.add_argument('-s', dest = 'nFramesStep', help = 'number of frames between each display', default = 1, type = int) 18 parser.add_argument('-s', dest = 'nFramesStep', help = 'number of frames between each display', default = 1, type = int)
19 parser.add_argument('--save-images', dest = 'saveAllImages', help = 'save all images', action = 'store_true')
20 parser.add_argument('--last-frame', dest = 'lastFrameNum', help = 'number of last frame number to save (for image saving, no display is made)', default = None, type = int)
19 21
20 args = parser.parse_args() 22 args = parser.parse_args()
21 23
22 homography = None 24 homography = None
23 if args.configFilename: # consider there is a configuration file 25 if args.configFilename: # consider there is a configuration file
37 if args.firstFrameNum != None: 39 if args.firstFrameNum != None:
38 firstFrameNum = args.firstFrameNum 40 firstFrameNum = args.firstFrameNum
39 41
40 objects = storage.loadTrajectoriesFromSqlite(databaseFilename, args.trajectoryType) 42 objects = storage.loadTrajectoriesFromSqlite(databaseFilename, args.trajectoryType)
41 boundingBoxes = storage.loadBoundingBoxTable(databaseFilename) 43 boundingBoxes = storage.loadBoundingBoxTable(databaseFilename)
42 cvutils.displayTrajectories(videoFilename, objects, boundingBoxes, homography, firstFrameNum, rescale = args.rescale, nFramesStep = args.nFramesStep) 44 cvutils.displayTrajectories(videoFilename, objects, boundingBoxes, homography, firstFrameNum, args.lastFrameNum, rescale = args.rescale, nFramesStep = args.nFramesStep, saveAllImages = args.saveAllImages)