goryhon commited on
Commit
e4fad93
·
verified ·
1 Parent(s): 9b3cfab

Update web-demos/hugging_face/track_anything.py

Browse files
web-demos/hugging_face/track_anything.py CHANGED
@@ -21,6 +21,7 @@ class TrackingAnything():
21
  masks = []
22
  logits = []
23
  painted_images = []
 
24
  for i in tqdm(range(len(images)), desc="Tracking image"):
25
  if i==0:
26
  mask, logit, painted_image, alpha_visual = self.cutie.track(images[i], template_mask)
 
21
  masks = []
22
  logits = []
23
  painted_images = []
24
+ alpha_visuals = []
25
  for i in tqdm(range(len(images)), desc="Tracking image"):
26
  if i==0:
27
  mask, logit, painted_image, alpha_visual = self.cutie.track(images[i], template_mask)