Mercurial Hosting > traffic-intelligence
diff scripts/extract-appearance-images.py @ 900:85b81c46c526
minor
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 21 Jun 2017 17:49:58 -0400 |
parents | |
children | 753a081989e2 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/scripts/extract-appearance-images.py Wed Jun 21 17:49:58 2017 -0400 @@ -0,0 +1,17 @@ +#! /usr/bin/env python + +import numpy as np +import argparse +from cv2 import SVM_RBF, SVM_C_SVC +#from cv2.ml import SVM_RBF, SVM_C_SVC, ROW_SAMPLE # row_sample for layout in cv2.ml.SVM_load + +import cvutils, moving, ml, storage + +parser = argparse.ArgumentParser(description='The program extracts labeled image patches to train the HoG-SVM classifier, and optionnally speed information') +parser.add_argument('-d', dest = 'directoryName', help = 'parent directory name for the directories containing the samples for the different road users', required = True) + +# need all info as for classification (image info) + +# todo speed info: distributions AND min speed equiprobable + +# provide csv delimiter for the classification file as arg