view scripts/create-bounding-boxes.py @ 1138:50a7e1b47a3c Tertuis-Ou-draogo/movingpy-issue-22-from-etienne-beauchamp-1584900380975

Close branch Tertuis-Ou-draogo/movingpy-issue-22-from-etienne-beauchamp-1584900380975.
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Sun, 22 Mar 2020 19:10:57 +0000
parents cc5cb04b04b0
children
line wrap: on
line source

#! /usr/bin/env python3

import argparse

from numpy.linalg.linalg import inv
from numpy import loadtxt

from trafficintelligence import storage

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.')
parser.add_argument('-d', dest = 'databaseFilename', help = 'name of the Sqlite database file', required = True)
parser.add_argument('-o', dest = 'homography', help = 'name of the image to world homography')

args = parser.parse_args()

homography = None
if args.homography is not None:
    homography = inv(loadtxt(args.homography))

storage.createBoundingBoxTable(args.databaseFilename, homography)