Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/extract-appearance-images.py @ 901:753a081989e2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
factorized some argument handling code
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 22 Jun 2017 12:02:34 -0400
parents 85b81c46c526
children c69a8defe5c3
comparison
equal deleted inserted replaced
900:85b81c46c526 901:753a081989e2
6 #from cv2.ml import SVM_RBF, SVM_C_SVC, ROW_SAMPLE # row_sample for layout in cv2.ml.SVM_load 6 #from cv2.ml import SVM_RBF, SVM_C_SVC, ROW_SAMPLE # row_sample for layout in cv2.ml.SVM_load
7 7
8 import cvutils, moving, ml, storage 8 import cvutils, moving, ml, storage
9 9
10 parser = argparse.ArgumentParser(description='The program extracts labeled image patches to train the HoG-SVM classifier, and optionnally speed information') 10 parser = argparse.ArgumentParser(description='The program extracts labeled image patches to train the HoG-SVM classifier, and optionnally speed information')
11 parser.add_argument('-d', dest = 'directoryName', help = 'parent directory name for the directories containing the samples for the different road users', required = True) 11 parser.add_argument('--cfg', dest = 'configFilename', help = 'name of the configuration file', required = True)
12
13 #parser.add_argument('-d', dest = 'directoryName', help = 'parent directory name for the directories containing the samples for the different road users', required = True)
14
15 args = parser.parse_args()
16 params = storage.ProcessParameters(args.configFilename)
17 classifierParams = storage.ClassifierParameters(params.classifierFilename)
12 18
13 # need all info as for classification (image info) 19 # need all info as for classification (image info)
14 20
15 # todo speed info: distributions AND min speed equiprobable 21 # todo speed info: distributions AND min speed equiprobable
16 22
17 # provide csv delimiter for the classification file as arg 23 # provide csv delimiter for the classification file as arg
24