Spaces:
Running
Running
Zhen Ye
commited on
Commit
·
b3371b1
1
Parent(s):
56e4f43
Suppress labels for non-relevant objects in visualization
Browse files- inference.py +6 -0
inference.py
CHANGED
|
@@ -1241,6 +1241,9 @@ def run_inference(
|
|
| 1241 |
display_boxes = np.array([d['bbox'] for d in dets])
|
| 1242 |
display_labels = []
|
| 1243 |
for d in dets:
|
|
|
|
|
|
|
|
|
|
| 1244 |
lbl = d.get('label', 'obj')
|
| 1245 |
# Append Track ID
|
| 1246 |
if 'track_id' in d:
|
|
@@ -2230,6 +2233,9 @@ def run_grounded_sam2_tracking(
|
|
| 2230 |
# Build enriched display labels
|
| 2231 |
display_labels = []
|
| 2232 |
for d in dets:
|
|
|
|
|
|
|
|
|
|
| 2233 |
lbl = d.get("label", "obj")
|
| 2234 |
if "track_id" in d:
|
| 2235 |
lbl = f"{d['track_id']} {lbl}"
|
|
|
|
| 1241 |
display_boxes = np.array([d['bbox'] for d in dets])
|
| 1242 |
display_labels = []
|
| 1243 |
for d in dets:
|
| 1244 |
+
if d.get("mission_relevant") is False:
|
| 1245 |
+
display_labels.append("")
|
| 1246 |
+
continue
|
| 1247 |
lbl = d.get('label', 'obj')
|
| 1248 |
# Append Track ID
|
| 1249 |
if 'track_id' in d:
|
|
|
|
| 2233 |
# Build enriched display labels
|
| 2234 |
display_labels = []
|
| 2235 |
for d in dets:
|
| 2236 |
+
if d.get("mission_relevant") is False:
|
| 2237 |
+
display_labels.append("")
|
| 2238 |
+
continue
|
| 2239 |
lbl = d.get("label", "obj")
|
| 2240 |
if "track_id" in d:
|
| 2241 |
lbl = f"{d['track_id']} {lbl}"
|