Mercurial Hosting > traffic-intelligence
diff scripts/merge-features.py @ 971:9897a13772fb
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 |
line wrap: on
line diff
--- a/scripts/merge-features.py Thu Dec 07 23:31:54 2017 -0500 +++ b/scripts/merge-features.py Wed Dec 13 14:06:20 2017 -0500 @@ -2,21 +2,23 @@ import sys, argparse, os.path, sqlite3 import cvutils, utils, moving, storage -from metadata import createDatabase, Site, VideoSequence, CameraView +from metadata import createDatabase, Site, VideoSequence, CameraView, getSite from datetime import datetime, timedelta +timeConverter = utils.TimeConverter() + parser = argparse.ArgumentParser(description='The program merges feature trajectories recorded from the same site synchronously between start and end time.') parser.add_argument('-i', dest = 'metadataFilename', help = 'name of the metadata file', required = True) parser.add_argument('-n', dest = 'siteId', help = 'site id or site name', required = True) -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 -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)') +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 +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) parser.add_argument('-o', dest = 'outputDBFilename', help = 'name of the output SQLite file', required = True) args = parser.parse_args() session = createDatabase(args.metadataFilename) -site = Site.getSite(session, args.siteId) +site = getSite(session, args.siteId) if site is None: print('Site {} was not found in {}. Exiting'.format(args.siteId, args.metadataFilename)) sys.exit()