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