Mercurial Hosting > traffic-intelligence
comparison 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 |
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] |