Mercurial > hg > nsaunier > 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 | 30711484d2f6 |
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 |