Mercurial Hosting > traffic-intelligence
comparison scripts/classify-objects.py @ 1169:9f7a4a026dab
correcting bug in classify objects that ignored the parameter for centile aggregation of speed, if used in classifier.cfg (updated pip version)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Mon, 19 Jul 2021 11:25:16 -0400 |
parents | 49c0bb6eacc2 |
children | d478d3122804 |
comparison
equal
deleted
inserted
replaced
1168:d71a4d174b1a | 1169:9f7a4a026dab |
---|---|
23 args = parser.parse_args() | 23 args = parser.parse_args() |
24 params, videoFilename, databaseFilename, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) | 24 params, videoFilename, databaseFilename, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) |
25 classifierParams = storage.ClassifierParameters(params.classifierFilename) | 25 classifierParams = storage.ClassifierParameters(params.classifierFilename) |
26 classifierParams.convertToFrames(params.videoFrameRate, 3.6) # conversion from km/h to m/frame | 26 classifierParams.convertToFrames(params.videoFrameRate, 3.6) # conversion from km/h to m/frame |
27 | 27 |
28 speedAggregationFunc = utils.aggregationFunction(classifierParams.speedAggregationMethod) | 28 speedAggregationFunc = utils.aggregationFunction(classifierParams.speedAggregationMethod, classifierParams.speedAggregationCentile) |
29 if speedAggregationFunc is None: | 29 if speedAggregationFunc is None: |
30 sys.exit() | 30 sys.exit() |
31 | 31 |
32 pedBikeCarSVM = ml.SVM_load(classifierParams.pedBikeCarSVMFilename) | 32 pedBikeCarSVM = ml.SVM_load(classifierParams.pedBikeCarSVMFilename) |
33 bikeCarSVM = ml.SVM_load(classifierParams.bikeCarSVMFilename) | 33 bikeCarSVM = ml.SVM_load(classifierParams.bikeCarSVMFilename) |