Mercurial Hosting > traffic-intelligence
comparison scripts/train-object-classification.py @ 923:238008f81c16
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 |