Spaces:
Paused
Paused
Update web-demos/hugging_face/tracker/base_tracker.py
Browse files
web-demos/hugging_face/tracker/base_tracker.py
CHANGED
|
@@ -56,7 +56,6 @@ class BaseTracker:
|
|
| 56 |
Input:
|
| 57 |
frames: numpy arrays (H, W, 3)
|
| 58 |
logit: numpy array (H, W), logit
|
| 59 |
-
|
| 60 |
Output:
|
| 61 |
mask: numpy arrays (H, W)
|
| 62 |
logit: numpy arrays, probability map (H, W)
|
|
@@ -85,7 +84,7 @@ class BaseTracker:
|
|
| 85 |
|
| 86 |
# map back
|
| 87 |
for k, v in self.mapper.remappings.items():
|
| 88 |
-
final_mask[out_mask == v] = k
|
| 89 |
|
| 90 |
num_objs = final_mask.max()
|
| 91 |
painted_image = frame
|
|
|
|
| 56 |
Input:
|
| 57 |
frames: numpy arrays (H, W, 3)
|
| 58 |
logit: numpy array (H, W), logit
|
|
|
|
| 59 |
Output:
|
| 60 |
mask: numpy arrays (H, W)
|
| 61 |
logit: numpy arrays, probability map (H, W)
|
|
|
|
| 84 |
|
| 85 |
# map back
|
| 86 |
for k, v in self.mapper.remappings.items():
|
| 87 |
+
final_mask[out_mask == v] = k
|
| 88 |
|
| 89 |
num_objs = final_mask.max()
|
| 90 |
painted_image = frame
|