Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/merge-features.py @ 971:9897a13772fb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added utils to load video sequence in metadata
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 13 Dec 2017 14:06:20 -0500
parents 75530d8c0090
children 7463c9bc846b
comparison
equal deleted inserted replaced
970:bf401567a933 971:9897a13772fb
1 #! /usr/bin/env python 1 #! /usr/bin/env python
2 2
3 import sys, argparse, os.path, sqlite3 3 import sys, argparse, os.path, sqlite3
4 import cvutils, utils, moving, storage 4 import cvutils, utils, moving, storage
5 from metadata import createDatabase, Site, VideoSequence, CameraView 5 from metadata import createDatabase, Site, VideoSequence, CameraView, getSite
6 from datetime import datetime, timedelta 6 from datetime import datetime, timedelta
7
8 timeConverter = utils.TimeConverter()
7 9
8 parser = argparse.ArgumentParser(description='The program merges feature trajectories recorded from the same site synchronously between start and end time.') 10 parser = argparse.ArgumentParser(description='The program merges feature trajectories recorded from the same site synchronously between start and end time.')
9 parser.add_argument('-i', dest = 'metadataFilename', help = 'name of the metadata file', required = True) 11 parser.add_argument('-i', dest = 'metadataFilename', help = 'name of the metadata file', required = True)
10 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)
11 parser.add_argument('-f', dest = 'startTime', help = 'time to start merging features (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)') # if not provided, take common time interval 13 parser.add_argument('-f', dest = 'startTime', help = 'time to start merging features (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)', type = timeConverter.convert) # if not provided, take common time interval
12 parser.add_argument('-l', dest = 'endTime', help = 'time to stop merging features (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)') 14 parser.add_argument('-l', dest = 'endTime', help = 'time to stop merging features (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)', type = timeConverter.convert)
13 parser.add_argument('-o', dest = 'outputDBFilename', help = 'name of the output SQLite file', required = True) 15 parser.add_argument('-o', dest = 'outputDBFilename', help = 'name of the output SQLite file', required = True)
14 16
15 args = parser.parse_args() 17 args = parser.parse_args()
16 18
17 session = createDatabase(args.metadataFilename) 19 session = createDatabase(args.metadataFilename)
18 20
19 site = Site.getSite(session, args.siteId) 21 site = getSite(session, args.siteId)
20 if site is None: 22 if site is None:
21 print('Site {} was not found in {}. Exiting'.format(args.siteId, args.metadataFilename)) 23 print('Site {} was not found in {}. Exiting'.format(args.siteId, args.metadataFilename))
22 sys.exit() 24 sys.exit()
23 else: 25 else:
24 site = site[0] 26 site = site[0]