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