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

web-demos/hugging_face/tracker/base_tracker.py

@@ -85,10 +85,7 @@ class BaseTracker:
         
         # map back
         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) 
+            final_mask[out_mask == v] = k 
 
         num_objs = final_mask.max()
         painted_image = frame
@@ -97,7 +94,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, alpha_visual
+        return final_mask, final_mask, painted_image
 
     @torch.no_grad()
     def clear_memory(self):