Mercurial Hosting > traffic-intelligence
comparison scripts/extract-appearance-images.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 | cc5cb04b04b0 |
children |
comparison
equal
deleted
inserted
replaced
1245:371c718e57d7 | 1246:2397de73770d |
---|---|
21 parser.add_argument('--prefix', dest = 'imagePrefix', help = 'image prefix', default = 'img') | 21 parser.add_argument('--prefix', dest = 'imagePrefix', help = 'image prefix', default = 'img') |
22 parser.add_argument('--ouput', dest = 'directoryName', help = 'parent directory name for the directories containing the samples for the different road users', default = '.') | 22 parser.add_argument('--ouput', dest = 'directoryName', help = 'parent directory name for the directories containing the samples for the different road users', default = '.') |
23 parser.add_argument('--compute-speed-distributions', dest = 'computeSpeedDistribution', help = 'computes the distribution of the road users of each type and fits parameters to each', action = 'store_true') | 23 parser.add_argument('--compute-speed-distributions', dest = 'computeSpeedDistribution', help = 'computes the distribution of the road users of each type and fits parameters to each', action = 'store_true') |
24 | 24 |
25 args = parser.parse_args() | 25 args = parser.parse_args() |
26 params, videoFilename, databaseFilename, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) | 26 params, videoFilename, databaseFilename, homography, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum = storage.processVideoArguments(args) |
27 classifierParams = storage.ClassifierParameters(params.classifierFilename) | 27 classifierParams = storage.ClassifierParameters(params.classifierFilename) |
28 | 28 |
29 classificationAnnotations = read_csv(args.classificationAnnotationFilename, index_col=0, delimiter = args.classificationAnnotationFilenameDelimiter, names = ["object_num", "road_user_type"]) | 29 classificationAnnotations = read_csv(args.classificationAnnotationFilename, index_col=0, delimiter = args.classificationAnnotationFilenameDelimiter, names = ["object_num", "road_user_type"]) |
30 annotatedObjectNumbers = classificationAnnotations.index.tolist() | 30 annotatedObjectNumbers = classificationAnnotations.index.tolist() |
31 | 31 |