Mercurial > hg > nsaunier > traffic-intelligence
comparison scripts/create-bounding-boxes.py @ 393:eaf7765221d9
added code to create bounding boxes and display them (inc scripts)
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Fri, 26 Jul 2013 02:12:08 -0400 |
| parents | |
| children | 6551a3cf1750 |
comparison
equal
deleted
inserted
replaced
| 392:dd4970f4221f | 393:eaf7765221d9 |
|---|---|
| 1 #! /usr/bin/env python | |
| 2 | |
| 3 import argparse | |
| 4 | |
| 5 import storage | |
| 6 | |
| 7 from numpy.linalg.linalg import inv | |
| 8 from numpy import loadtxt | |
| 9 | |
| 10 parser = argparse.ArgumentParser(description='The program creates bounding boxes in image space around all features (for display and for comparison to ground truth in the form of bouding boxes.') | |
| 11 parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True) | |
| 12 parser.add_argument('-o', dest = 'homography', help = 'name of the image to world homography') | |
| 13 | |
| 14 args = parser.parse_args() | |
| 15 | |
| 16 homography = None | |
| 17 if args.homography != None: | |
| 18 homography = inv(loadtxt(args.homography)) | |
| 19 | |
| 20 storage.createBoundingBoxTable(args.databaseFilename, homography) | |
| 21 |