Mercurial > hg > nsaunier > traffic-intelligence

comparison 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
comparison
equal deleted inserted replaced
922:acb5379c5fd7 923:238008f81c16
43 trainingSamplesPV = {} 43 trainingSamplesPV = {}
44 trainingLabelsPV = {} 44 trainingLabelsPV = {}
45 45
46 for k, v in imageDirectories.iteritems(): 46 for k, v in imageDirectories.iteritems():
47 print('Loading {} samples'.format(k)) 47 print('Loading {} samples'.format(k))
48 trainingSamples, trainingLabels = cvutils.createHOGTrainingSet(v, moving.userType2Num[k], classifierParams.hogRescaleSize, classifierParams.hogNOrientations, classifierParams.hogNPixelsPerCell, classifierParams.hogNCellsPerBlock) 48 trainingSamples, trainingLabels = cvutils.createHOGTrainingSet(v, moving.userType2Num[k], classifierParams.hogRescaleSize, classifierParams.hogNOrientations, classifierParams.hogNPixelsPerCell, classifierParams.hogBlockNorm, classifierParams.hogNCellsPerBlock)
49 trainingSamplesPBV[k], trainingLabelsPBV[k] = trainingSamples, trainingLabels 49 trainingSamplesPBV[k], trainingLabelsPBV[k] = trainingSamples, trainingLabels
50 if k != 'pedestrian': 50 if k != 'pedestrian':
51 trainingSamplesBV[k], trainingLabelsBV[k] = trainingSamples, trainingLabels 51 trainingSamplesBV[k], trainingLabelsBV[k] = trainingSamples, trainingLabels
52 if k != 'car': 52 if k != 'car':
53 trainingSamplesPB[k], trainingLabelsPB[k] = trainingSamples, trainingLabels 53 trainingSamplesPB[k], trainingLabelsPB[k] = trainingSamples, trainingLabels