Mercurial Hosting > traffic-intelligence
comparison scripts/extract-appearance-images.py @ 901:753a081989e2
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 |