Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/compute-clearmot.py @ 1158:7eb972942f22

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