Skip to content

Commit 7a4236a

Browse files
Fix indexing bug in prompt generator
1 parent f351d72 commit 7a4236a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

micro_sam/prompt_generators.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def __call__(self, gt, gt_id, center_coordinates, bbox_coordinates):
3232
if self.get_box_prompts:
3333
bbox_list = [bbox_coordinates]
3434

35-
object_mask = gt == gt_id + 1 # alloting a label id to obtain the coordinates of desired seeds
35+
object_mask = gt == gt_id
3636

3737
# getting the additional positive points by randomly sampling points from this mask
3838
n_positive_remaining = self.n_positive_points - 1

0 commit comments

Comments
 (0)