Mercurial Hosting > traffic-intelligence
diff scripts/train-object-classification.py @ 1226:d478d3122804
change of bicycle to cyclist
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Wed, 12 Jul 2023 12:12:37 -0400 |
| parents | 933670761a57 |
| children | 5654c9173548 |
line wrap: on
line diff
--- a/scripts/train-object-classification.py Mon Jul 03 10:33:20 2023 -0400 +++ b/scripts/train-object-classification.py Wed Jul 12 12:12:37 2023 -0400 @@ -23,7 +23,7 @@ classifierParams = storage.ClassifierParameters(args.configFilename) imageDirectories = {'pedestrian': args.directoryName + "/Pedestrians/", - 'bicycle': args.directoryName + "/Cyclists/", + 'cyclist': args.directoryName + "/Cyclists/", 'car': args.directoryName + "/Vehicles/"} trainingSamplesPBV = {} @@ -43,7 +43,7 @@ trainingSamplesBV[k], trainingLabelsBV[k] = trainingSamples, trainingLabels if k != 'car': trainingSamplesPB[k], trainingLabelsPB[k] = trainingSamples, trainingLabels - if k != 'bicycle': + if k != 'cyclist': trainingSamplesPV[k], trainingLabelsPV[k] = trainingSamples, trainingLabels # Training the Support Vector Machine