Mercurial > hg > nsaunier > traffic-intelligence
comparison scripts/classify-objects.py @ 1246:2397de73770d
dltrack saves after projecting coordinates
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Fri, 09 Feb 2024 17:47:33 -0500 |
| parents | 00b71da2baac |
| children |
comparison
equal
deleted
inserted
replaced
| 1245:371c718e57d7 | 1246:2397de73770d |
|---|---|
| 27 parser.add_argument('--plot-speed-distributions', dest = 'plotSpeedDistribution', help = 'simply plots the distributions used for each user type', action = 'store_true') | 27 parser.add_argument('--plot-speed-distributions', dest = 'plotSpeedDistribution', help = 'simply plots the distributions used for each user type', action = 'store_true') |
| 28 parser.add_argument('--max-speed-distribution-plot', dest = 'maxSpeedDistributionPlot', help = 'if plotting the user distributions, the maximum speed to display (km/h)', type = float, default = 50.) | 28 parser.add_argument('--max-speed-distribution-plot', dest = 'maxSpeedDistributionPlot', help = 'if plotting the user distributions, the maximum speed to display (km/h)', type = float, default = 50.) |
| 29 parser.add_argument('--verbose', dest = 'verbose', help = 'verbose information', action = 'store_true') | 29 parser.add_argument('--verbose', dest = 'verbose', help = 'verbose information', action = 'store_true') |
| 30 | 30 |
| 31 args = parser.parse_args() | 31 args = parser.parse_args() |
| 32 params, videoFilename, databaseFilename, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) | 32 params, videoFilename, databaseFilename, homography, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) |
| 33 classifierParams = storage.ClassifierParameters(params.classifierFilename) | 33 classifierParams = storage.ClassifierParameters(params.classifierFilename) |
| 34 classifierParams.convertToFrames(params.videoFrameRate, 3.6) # conversion from km/h to m/frame | 34 classifierParams.convertToFrames(params.videoFrameRate, 3.6) # conversion from km/h to m/frame |
| 35 | 35 |
| 36 speedAggregationFunc = utils.aggregationFunction(classifierParams.speedAggregationMethod, classifierParams.speedAggregationCentile) | 36 speedAggregationFunc = utils.aggregationFunction(classifierParams.speedAggregationMethod, classifierParams.speedAggregationCentile) |
| 37 if speedAggregationFunc is None: | 37 if speedAggregationFunc is None: |