Spaces:
Paused
Paused
Zhen Ye Claude Opus 4.6 commited on
Commit ·
04c92f3
1
Parent(s): c97a5f9
fix: hide instance_id from segmentation overlay labels
Browse filesDisplay 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]>
- 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 =
|
| 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)
|