Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/classify-objects.py @ 1169:9f7a4a026dab

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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)