Update web-demos/hugging_face/track_anything.py

web-demos/hugging_face/track_anything.py

@@ -23,16 +23,18 @@ class TrackingAnything():
         painted_images = []
         for i in tqdm(range(len(images)), desc="Tracking image"):
             if i==0:           
-                mask, logit, painted_image = self.cutie.track(images[i], template_mask)
+                mask, logit, painted_image, alpha_visual = self.cutie.track(images[i], template_mask)
+                alpha_visuals.append(alpha_visual)
                 masks.append(mask)
                 logits.append(logit)
                 painted_images.append(painted_image)
             else:
-                mask, logit, painted_image = self.cutie.track(images[i])
+                mask, logit, painted_image, alpha_visual = self.cutie.track(images[i])
                 masks.append(mask)
                 logits.append(logit)
-                painted_images.append(painted_image)  
-        return masks, logits, painted_images
+                painted_images.append(painted_image)
+                alpha_visuals.append(alpha_visual)
+        return masks, logits, painted_images, alpha_visuals