Mercurial Hosting > traffic-intelligence
view scripts/init_tracking.py @ 923:238008f81c16
corrected bug
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 05 Jul 2017 23:01:24 -0400 |
parents | 2faabcbde2c4 |
children |
line wrap: on
line source
#! /usr/bin/env python import sys, argparse, os.path, storage, utils from shutil import copy 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') if os.path.exists(directoryName+'/../classifier.cfg') and not os.path.exists('./classifier.cfg'): copy(directoryName+'/../classifier.cfg', 'classifier.cfg') print('Configuration file classifier.cfg successfully copied to the current directory') # 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')