Mercurial Hosting > traffic-intelligence
view scripts/init_tracking.py @ 747:d45ab817ee11 dev
added script to initialize tracking (copying and setting up basic files)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 02 Oct 2015 11:29:43 -0400 |
parents | |
children | 2faabcbde2c4 |
line wrap: on
line source
#! /usr/bin/env python import sys, argparse, os.path, storage, utils from cvutils import getImagesFromVideo from matplotlib.pyplot import imsave # could try to guess the video # check if there is already a tracking.cfg file parser = argparse.ArgumentParser(description='The program initilizes the files for tracking: copy tracking.cfg, sets up with the video filename, generates a frame image (frame.png) and prints the next commands') parser.add_argument('-i', dest = 'videoFilename', help = 'filename of the video sequence', required = True) args = parser.parse_args() # assumes tracking.cfg is in the parent directory to the directory of the traffic intelligence python modules matchingPaths = [s for s in sys.path if 'traffic-intelligence' in s] #if len(matchingPaths) > 1: # print('Too many matching paths for Traffic Intelligence modules: {}'.format(matchingPaths)) if len(matchingPaths) == 0: print('No environment path to Traffic Intelligence modules.\nExiting') sys.exit() else: directoryName = matchingPaths[0] if directoryName.endswith('/'): directoryName = directoryName[:-1] if os.path.exists(directoryName+'/../tracking.cfg') and not os.path.exists('./tracking.cfg'): f = storage.openCheck(directoryName+'/../tracking.cfg') out = storage.openCheck('./tracking.cfg', 'w') for l in f: if 'video-filename' in l: tmp = l.split('=') out.write(tmp[0]+'= '+args.videoFilename+'\n') elif 'database-filename' in l: tmp = l.split('=') out.write(tmp[0]+'= '+utils.removeExtension(args.videoFilename)+'.sqlite\n') else: out.write(l) f.close() out.close() print('Configuration file tracking.cfg successfully copied to the current directory with video and database filename adapted') # extract image from video image = getImagesFromVideo(args.videoFilename, saveImage = True, outputPrefix = 'frame') print('first video frame successfully copied to the current directory') # next commands print('--------------------------------------\nHere are a sample of the next command to compute the homography,\ntrack features, group them in objects and display object trajectories\n--------------------------------------') print('compute_homography -i [frame.png] -w [world_image] -n [npoints] -u [unit_per_pixel]') print('feature-based-tracking tracking.cfg --tf') print('feature-based-tracking tracking.cfg --gf') print('display-trajectories --cfg tracking.cfg -t object')