Skip to content

Commit b44bf6f

Browse files
Apply fix for SAM3 segmentation post-processing
1 parent 14b27a9 commit b44bf6f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

inference/models/sam3/segment_anything3.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,8 @@ def segment_image(
438438
post = PostProcessImage(
439439
max_dets_per_img=-1,
440440
iou_type="segm",
441-
use_original_sizes=True,
441+
use_original_sizes_box=True,
442+
use_original_sizes_mask=True,
442443
convert_mask_to_rle=False,
443444
detection_threshold=float(
444445
output_prob_thresh if output_prob_thresh is not None else 0.35

0 commit comments

Comments
 (0)