Zhen Ye commited on
Commit
b3371b1
·
1 Parent(s): 56e4f43

Suppress labels for non-relevant objects in visualization

Browse files
Files changed (1) hide show
  1. 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}"