Mercurial Hosting > traffic-intelligence
changeset 923:238008f81c16
corrected bug
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 05 Jul 2017 23:01:24 -0400 |
parents | acb5379c5fd7 |
children | a71455bd8367 |
files | scripts/train-object-classification.py |
diffstat | 1 files changed, 1 insertions(+), 1 deletions(-) [+] |
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