Mercurial Hosting > traffic-intelligence
changeset 900:85b81c46c526
minor
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 21 Jun 2017 17:49:58 -0400 |
parents | 1466a63dd1cf |
children | 753a081989e2 |
files | scripts/extract-appearance-images.py scripts/train-object-classification.py |
diffstat | 2 files changed, 18 insertions(+), 1 deletions(-) [+] |
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
--- a/scripts/train-object-classification.py Wed Jun 21 17:44:09 2017 -0400 +++ b/scripts/train-object-classification.py Wed Jun 21 17:49:58 2017 -0400 @@ -1,7 +1,7 @@ #! /usr/bin/env python import numpy as np -import sys, argparse +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