Mercurial Hosting > traffic-intelligence
diff scripts/compute-clearmot.py @ 1249:2aa56b101041
added mask functionality for dltrack
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 15 Feb 2024 14:09:52 -0500 |
parents | b684135d817f |
children |
line wrap: on
line diff
--- a/scripts/compute-clearmot.py Thu Feb 15 14:09:23 2024 -0500 +++ b/scripts/compute-clearmot.py Thu Feb 15 14:09:52 2024 -0500 @@ -45,9 +45,9 @@ if args.maskFilename is not None: maskObjects = [] from matplotlib.pyplot import imread - mask = imread(args.maskFilename) - if len(mask) > 1: - mask = mask[:,:,0] + mask = imread(args.maskFilename, cv2.IMREAD_GRAYSCALE) + #if len(mask) > 1: if loaded as RGB + # mask = mask[:,:,0] for obj in objects: maskObjects += obj.getObjectsInMask(mask, invHomography, 10) # TODO add option to keep object if at least one feature in mask objects = maskObjects