Mercurial Hosting > traffic-intelligence
comparison scripts/classify-objects.py @ 522:ce40a89bd6ae
added functions for classification refactored from Sohail s work for TRB/TRC (to be tested)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 18 Jun 2014 23:36:56 -0400 |
parents | |
children | f527159815ab |
comparison
equal
deleted
inserted
replaced
521:3707eeb20f25 | 522:ce40a89bd6ae |
---|---|
1 #! /usr/bin/env python | |
2 | |
3 import numpy as np | |
4 import sys, argparse | |
5 from cv2 import SVM_RBF, SVM_C_SVC | |
6 | |
7 import cvutils, moving, ml | |
8 | |
9 parser = argparse.ArgumentParser(description='The program processes indicators for all pairs of road users in the scene') | |
10 parser.add_argument('-d', dest = 'directoryName', help = 'name of the parent directory containing the videos and extracted trajectories to process', required = True) | |
11 #parser.add_argument('--kernel', dest = 'kernelType', help = 'kernel type for the support vector machine (SVM)', default = SVM_RBF, type = long) | |
12 #parser.add_argument('--svm', dest = 'svmType', help = 'SVM type', default = SVM_C_SVC, type = long) | |
13 #parser.add_argument('-s', dest = 'rescaleSize', help = 'rescale size of image samples', default = 64, type = int) | |
14 #parser.add_argument('-o', dest = 'nOrientations', help = 'number of orientations in HoG', default = 9, type = int) | |
15 #parser.add_argument('-p', dest = 'nPixelsPerCell', help = 'number of pixels per cell', default = 8, type = int) | |
16 #parser.add_argument('-c', dest = 'nCellsPerBlock', help = 'number of cells per block', default = 2, type = int) | |
17 args = parser.parse_args() | |
18 | |
19 print('TODO') # add all parameters for classification (distribution parameters) |