Mercurial Hosting > traffic-intelligence
comparison scripts/compute-clearmot.py @ 1158:7eb972942f22
solving bug to pass kwargs for plotting
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Tue, 05 Jan 2021 10:45:39 -0500 |
parents | cc5cb04b04b0 |
children | d24d57e4de24 |
comparison
equal
deleted
inserted
replaced
1157:173b7926734e | 1158:7eb972942f22 |
---|---|
46 mask = imread(args.maskFilename) | 46 mask = imread(args.maskFilename) |
47 if len(mask) > 1: | 47 if len(mask) > 1: |
48 mask = mask[:,:,0] | 48 mask = mask[:,:,0] |
49 for obj in objects: | 49 for obj in objects: |
50 maskObjects += obj.getObjectsInMask(mask, invHomography, 10) # TODO add option to keep object if at least one feature in mask | 50 maskObjects += obj.getObjectsInMask(mask, invHomography, 10) # TODO add option to keep object if at least one feature in mask |
51 objects = maskObjects | 51 objects = maskObjects |
52 | 52 |
53 annotations = storage.loadBBMovingObjectsFromSqlite(args.groundTruthDatabaseFilename) | 53 annotations = storage.loadBBMovingObjectsFromSqlite(args.groundTruthDatabaseFilename) |
54 for a in annotations: | 54 for a in annotations: |
55 a.computeCentroidTrajectory(invHomography) | 55 a.computeCentroidTrajectory(invHomography) |
56 | 56 |