Update web-demos/hugging_face/tracker/base_tracker.py

web-demos/hugging_face/tracker/base_tracker.py

@@ -87,6 +87,9 @@ class BaseTracker:
         for k, v in self.mapper.remappings.items():
             final_mask[out_mask == v] = k
 
+        alpha_mask = np.where(final_mask > 0, 255, 0).astype(np.uint8)
+        alpha_visual = np.stack([alpha_mask]*3 + [alpha_mask], axis=-1) 
+
         num_objs = final_mask.max()
         painted_image = frame
         for obj in range(1, num_objs+1):
@@ -94,7 +97,7 @@ class BaseTracker:
                 continue
             painted_image = mask_painter(painted_image, (final_mask==obj).astype('uint8'), mask_color=obj+1)
 
-        return final_mask, final_mask, painted_image
+        return final_mask, final_mask, painted_image, alpha_visual
 
     @torch.no_grad()
     def clear_memory(self):