Mercurial > hg > nsaunier > traffic-intelligence
comparison scripts/classify-objects.py @ 614:5e09583275a4
Merged Nicolas/trafficintelligence into default
| author | Mohamed Gomaa <eng.m.gom3a@gmail.com> |
|---|---|
| date | Fri, 05 Dec 2014 12:13:53 -0500 |
| parents | 95276d310972 |
| children | da1352b89d02 |
comparison
equal
deleted
inserted
replaced
| 598:11f96bd08552 | 614:5e09583275a4 |
|---|---|
| 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('--cfg', dest = 'configFilename', help = 'name of the configuration file', required = True) | |
| 11 parser.add_argument('-d', dest = 'directoryName', help = 'name of the parent directory containing the videos and extracted trajectories to process', required = True) | |
| 12 # parser.add_argument('-o', dest = 'homographyFilename', help = 'name of the image to world homography file') | |
| 13 # need a classification config file for speed distribution parameters, svm models, frequency parameters, area parameters etc | |
| 14 #parser.add_argument('--cfg', dest = 'svmType', help = 'SVM type', default = SVM_C_SVC, type = long) | |
| 15 | |
| 16 | |
| 17 #parser.add_argument('-s', dest = 'rescaleSize', help = 'rescale size of image samples', default = 64, type = int) | |
| 18 #parser.add_argument('-o', dest = 'nOrientations', help = 'number of orientations in HoG', default = 9, type = int) | |
| 19 #parser.add_argument('-p', dest = 'nPixelsPerCell', help = 'number of pixels per cell', default = 8, type = int) | |
| 20 #parser.add_argument('-c', dest = 'nCellsPerBlock', help = 'number of cells per block', default = 2, type = int) | |
| 21 | |
| 22 args = parser.parse_args() | |
| 23 params = storage.ProcessParameters(args.configFilename) |