github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(backend): issue w/ multiple bbox and sam1

Browse Source
This commit is contained in:
psychedelicious
2025-09-10 19:21:59 +10:00
parent 9e4d441e2e
commit ec1a058dbe
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/image_util/segment_anything/segment_anything_pipeline.py
View File

@@ -49,7 +49,7 @@ class SegmentAnythingPipeline(RawModel):
torch.Tensor: The segmentation masks. dtype: torch.bool. shape: [num_masks, channels, height, width].
"""
input_boxes: list[list[list[float]]] = []
input_boxes: list[list[float]] = []
input_points: list[list[list[float]]] = []
input_labels: list[list[int]] = []
@@ -74,7 +74,7 @@ class SegmentAnythingPipeline(RawModel):
labels.append(point.label.value)
if box is not None:
input_boxes.append([box])
input_boxes.append(box)
if points is not None:
input_points.append(points)
if labels is not None:
@@ -82,7 +82,7 @@ class SegmentAnythingPipeline(RawModel):
processed_inputs = self._sam_processor(
images=image,
input_boxes=input_boxes if input_boxes else None,
input_boxes=[input_boxes] if input_boxes else None,
input_points=input_points if input_points else None,
input_labels=input_labels if input_labels else None,
return_tensors="pt",