comparison scripts/merge-features.py @ 985:668a85c963c3

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
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, getSite 5 from metadata import connectDatabase, Site, VideoSequence, CameraView, getSite
6 from datetime import datetime, timedelta 6 from datetime import datetime, timedelta
7 7
8 timeConverter = utils.TimeConverter() 8 timeConverter = utils.TimeConverter()
9 9
10 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.')
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) 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)
15 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)
16 16
17 args = parser.parse_args() 17 args = parser.parse_args()
18 18
19 session = createDatabase(args.metadataFilename) 19 session = connectDatabase(args.metadataFilename)
20 20
21 site = getSite(session, args.siteId) 21 site = getSite(session, args.siteId)
22 if site is None: 22 if site is None:
23 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))
24 sys.exit() 24 sys.exit()