Zhen Ye Claude Opus 4.6 commited on
Commit
04c92f3
·
1 Parent(s): c97a5f9

fix: hide instance_id from segmentation overlay labels

Browse files

Display only the class name on bounding boxes instead of
"instance_id class_name". The instance_id is still preserved
in job detection data.

Co-Authored-By: Claude Opus 4.6 <[email protected]>

Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -1233,7 +1233,7 @@ def _gsam2_render_frame(
1233
 
1234
  for _obj_id, obj_info in frame_objects.items():
1235
  mask = obj_info.mask
1236
- label = f"{obj_info.instance_id} {obj_info.class_name}"
1237
  if mask is not None:
1238
  if isinstance(mask, torch.Tensor):
1239
  mask_np = mask.cpu().numpy().astype(bool)
 
1233
 
1234
  for _obj_id, obj_info in frame_objects.items():
1235
  mask = obj_info.mask
1236
+ label = obj_info.class_name
1237
  if mask is not None:
1238
  if isinstance(mask, torch.Tensor):
1239
  mask_np = mask.cpu().numpy().astype(bool)