Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/classify-objects.py @ 1028:cc5cb04b04b0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
major update using the trafficintelligence package name and install through pip
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 15 Jun 2018 11:19:10 -0400
parents a13f47c8931d
children 49c0bb6eacc2
comparison
equal deleted inserted replaced
1027:6129296848d3 1028:cc5cb04b04b0
1 #! /usr/bin/env python3 1 #! /usr/bin/env python3
2 2
3 import cvutils, moving, ml, storage, utils 3 import sys, argparse
4 4
5 import numpy as np 5 import numpy as np
6 import sys, argparse
7 #from cv2.ml import SVM_RBF, SVM_C_SVC
8 import cv2 6 import cv2
9 from scipy.stats import norm, lognorm 7 from scipy.stats import norm, lognorm
8
9 from trafficintelligence import cvutils, moving, ml, storage, utils
10 10
11 # TODO add mode detection live, add choice of kernel and svm type (to be saved in future classifier format) 11 # TODO add mode detection live, add choice of kernel and svm type (to be saved in future classifier format)
12 12
13 parser = argparse.ArgumentParser(description='The program processes indicators for all pairs of road users in the scene') 13 parser = argparse.ArgumentParser(description='The program processes indicators for all pairs of road users in the scene')
14 parser.add_argument('--cfg', dest = 'configFilename', help = 'name of the configuration file', required = True) 14 parser.add_argument('--cfg', dest = 'configFilename', help = 'name of the configuration file', required = True)