Mercurial Hosting > traffic-intelligence

comparison scripts/display-synced-trajectories.py @ 985:668a85c963c3

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
work on processing and managing large video datasets
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 06 Mar 2018 22:44:33 -0500
parents 7463c9bc846b
children 933670761a57
comparison
equal deleted inserted replaced
984:a69695d14e59 985:668a85c963c3
3 import sys, argparse, os.path 3 import sys, argparse, os.path
4 from datetime import datetime, timedelta 4 from datetime import datetime, timedelta
5 import numpy as np 5 import numpy as np
6 import cv2 6 import cv2
7 import cvutils, utils, storage 7 import cvutils, utils, storage
8 from metadata import createDatabase, Site, CameraView, VideoSequence 8 from metadata import connectDatabase, Site, CameraView, VideoSequence
9 9
10 parser = argparse.ArgumentParser(description='The program displays several views of the same site synchronously.') 10 parser = argparse.ArgumentParser(description='The program displays several views of the same site synchronously.')
11 parser.add_argument('--db', dest = 'metadataFilename', help = 'name of the metadata file', required = True) 11 parser.add_argument('--db', dest = 'metadataFilename', help = 'name of the metadata file', required = True)
12 #parser.add_argument('-n', dest = 'siteId', help = 'site id or site name', required = True) 12 #parser.add_argument('-n', dest = 'siteId', help = 'site id or site name', required = True)
13 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True) 13 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True)
17 parser.add_argument('-s', dest = 'step', help = 'display every s image', default = 1, type = int) 17 parser.add_argument('-s', dest = 'step', help = 'display every s image', default = 1, type = int)
18 parser.add_argument('-u', dest = 'undistort', help = 'undistort the video (because features have been extracted that way)', action = 'store_true') 18 parser.add_argument('-u', dest = 'undistort', help = 'undistort the video (because features have been extracted that way)', action = 'store_true')
19 19
20 args = parser.parse_args() 20 args = parser.parse_args()
21 21
22 session = createDatabase(args.metadataFilename) 22 session = connectDatabase(args.metadataFilename)
23 23
24 mergedSequence = session.query(VideoSequence).filter(VideoSequence.databaseFilename == args.databaseFilename).first() 24 mergedSequence = session.query(VideoSequence).filter(VideoSequence.databaseFilename == args.databaseFilename).first()
25 if mergedSequence is None: 25 if mergedSequence is None:
26 print('Video sequence {} was not found in {}. Exiting'.format(args.databaseFilename, args.metadataFilename)) 26 print('Video sequence {} was not found in {}. Exiting'.format(args.databaseFilename, args.metadataFilename))
27 sys.exit() 27 sys.exit()