Mercurial Hosting > traffic-intelligence

diff scripts/train-object-classification.py @ 923:238008f81c16

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected bug
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 05 Jul 2017 23:01:24 -0400
parents 85b81c46c526
children ec1682ed999f
line wrap: on
line diff
--- a/scripts/train-object-classification.py	Wed Jul 05 18:07:53 2017 -0400
+++ b/scripts/train-object-classification.py	Wed Jul 05 23:01:24 2017 -0400
@@ -45,7 +45,7 @@
 
 for k, v in imageDirectories.iteritems():
     print('Loading {} samples'.format(k))
-    trainingSamples, trainingLabels = cvutils.createHOGTrainingSet(v, moving.userType2Num[k], classifierParams.hogRescaleSize, classifierParams.hogNOrientations, classifierParams.hogNPixelsPerCell, classifierParams.hogNCellsPerBlock)
+    trainingSamples, trainingLabels = cvutils.createHOGTrainingSet(v, moving.userType2Num[k], classifierParams.hogRescaleSize, classifierParams.hogNOrientations, classifierParams.hogNPixelsPerCell, classifierParams.hogBlockNorm, classifierParams.hogNCellsPerBlock)
     trainingSamplesPBV[k], trainingLabelsPBV[k] = trainingSamples, trainingLabels
     if k != 'pedestrian':
 	trainingSamplesBV[k], trainingLabelsBV[k] = trainingSamples, trainingLabels