Mercurial Hosting > traffic-intelligence
diff scripts/train-object-classification.py @ 1228:5654c9173548
merged (bicycle)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 12 Jul 2023 13:21:08 -0400 |
parents | d478d3122804 |
children |
line wrap: on
line diff
--- a/scripts/train-object-classification.py Wed Jul 12 13:14:09 2023 -0400 +++ b/scripts/train-object-classification.py Wed Jul 12 13:21:08 2023 -0400 @@ -22,9 +22,9 @@ args = parser.parse_args() classifierParams = storage.ClassifierParameters(args.configFilename) -imageDirectories = {'pedestrian': args.directoryName + "/Pedestrians/", - 'bicycle': args.directoryName + "/Cyclists/", - 'car': args.directoryName + "/Vehicles/"} +imageDirectories = {moving.userTypeNames[2]: args.directoryName + "/Pedestrians/", + moving.userTypeNames[4]: args.directoryName + "/Cyclists/", + moving.userTypeNames[1]: args.directoryName + "/Vehicles/"} trainingSamplesPBV = {} trainingLabelsPBV = {} @@ -39,11 +39,11 @@ print('Loading {} samples'.format(k)) 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': + if k != moving.userTypeNames[2]: trainingSamplesBV[k], trainingLabelsBV[k] = trainingSamples, trainingLabels - if k != 'car': + if k != moving.userTypeNames[1]: trainingSamplesPB[k], trainingLabelsPB[k] = trainingSamples, trainingLabels - if k != 'bicycle': + if k != moving.userTypeNames[4]: trainingSamplesPV[k], trainingLabelsPV[k] = trainingSamples, trainingLabels # Training the Support Vector Machine