Mercurial Hosting > traffic-intelligence
diff scripts/create-metadata.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 | 5d788d2e8ffc |
children | 933670761a57 |
line wrap: on
line diff
--- a/scripts/create-metadata.py Thu Dec 07 23:31:54 2017 -0500 +++ b/scripts/create-metadata.py Wed Dec 13 14:06:20 2017 -0500 @@ -1,14 +1,30 @@ #! /usr/bin/env python -import sys, argparse -import cvutils +import argparse +from datetime import datetime +import metadata, utils +timeConverter = utils.TimeConverter() -parser = argparse.ArgumentParser(description='The program displays the video.') +parser = argparse.ArgumentParser(description='The program add camera views (metadata.CameraView) for a site or video sequences (metadata.VideoSequence) for a site and a view.') #parser.add_argument('-d', dest = 'siteDirectory', help = 'name of the directory for the site')#, required = True -parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the metadata filename') -parser.add_argument('-s', dest = 'site', help = 'site id') - +parser.add_argument('-i', dest = 'databaseFilename', help = 'name of the metadata filename', required = True) +parser.add_argument('-d', dest = 'dirname', help = 'directory name containing sites or video sequences for a given view') +#parser.add_argument('-s', dest = 'siteId', help = 'site id (if provided, the program adds video sequences for the camera view)') +parser.add_argument('-v', dest = 'viewId', help = 'camera view id') +parser.add_argument('--nviews', dest = 'nViewsPerSite', help = 'default number of camera views', type = int, default = 1) +parser.add_argument('-s', dest = 'startTime', help = 'starting time of the first video (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)', type = timeConverter.convert) +parser.add_argument('--timeformat', dest = 'timeFormat', help = 'time format of the video filenames (optional) (eg %%Y_%%m%%d_%%H%%M%%S, eg 2017_0627_163231)') args = parser.parse_args() -print('Unfinished, look at methods in metadata module') +session = metadata.createDatabase(args.databaseFilename) +if args.viewId is not None: + # sites = metadata.getSite(session, args.siteId) + # if len(sites) > 1: + # print('{} sites found matching {}, using the first {}'.format(len(sites), args.siteId, sites[0].name)) + # site = sites[0] + cameraView = metadata.getCameraView(session, args.viewId) + metadata.initializeVideos(session, cameraView, args.dirname, args.startTime, args.timeFormat) +else: + metadata.initializeSites(session, args.dirname, args.nViewsPerSite) +