Mercurial Hosting > traffic-intelligence

comparison scripts/create-metadata.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 5d788d2e8ffc
children 933670761a57
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 3 import argparse
4 import cvutils 4 from datetime import datetime
5 import metadata, utils
5 6
7 timeConverter = utils.TimeConverter()
6 8
7 parser = argparse.ArgumentParser(description='The program displays the video.') 9 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.')
8 #parser.add_argument('-d', dest = 'siteDirectory', help = 'name of the directory for the site')#, required = True 10 #parser.add_argument('-d', dest = 'siteDirectory', help = 'name of the directory for the site')#, required = True
9 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the metadata filename') 11 parser.add_argument('-i', dest = 'databaseFilename', help = 'name of the metadata filename', required = True)
10 parser.add_argument('-s', dest = 'site', help = 'site id') 12 parser.add_argument('-d', dest = 'dirname', help = 'directory name containing sites or video sequences for a given view')
11 13 #parser.add_argument('-s', dest = 'siteId', help = 'site id (if provided, the program adds video sequences for the camera view)')
14 parser.add_argument('-v', dest = 'viewId', help = 'camera view id')
15 parser.add_argument('--nviews', dest = 'nViewsPerSite', help = 'default number of camera views', type = int, default = 1)
16 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)
17 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)')
12 args = parser.parse_args() 18 args = parser.parse_args()
13 19
14 print('Unfinished, look at methods in metadata module') 20 session = metadata.createDatabase(args.databaseFilename)
21 if args.viewId is not None:
22 # sites = metadata.getSite(session, args.siteId)
23 # if len(sites) > 1:
24 # print('{} sites found matching {}, using the first {}'.format(len(sites), args.siteId, sites[0].name))
25 # site = sites[0]
26 cameraView = metadata.getCameraView(session, args.viewId)
27 metadata.initializeVideos(session, cameraView, args.dirname, args.startTime, args.timeFormat)
28 else:
29 metadata.initializeSites(session, args.dirname, args.nViewsPerSite)
30