Skip to content

Commit 5d6af10

Browse files
Fix typo
1 parent 06bdcbe commit 5d6af10

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

micro_sam/prompt_generators.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,10 +115,10 @@ def __call__(
115115
# getting the negative points
116116
# for this we do the opposite and we set the mask to the bounding box - the object mask
117117
# we need to dilate the object mask before doing this: we use scipy.ndimage.binary_dilation for this
118-
dilated_object = binary_dilation(object_mask, iterations=self.dilation_strensegmentationh)
118+
dilated_object = binary_dilation(object_mask, iterations=self.dilation_strength)
119119
background_mask = np.zeros(segmentation.shape)
120120
background_mask[bbox_coordinates[0]:bbox_coordinates[2], bbox_coordinates[1]:bbox_coordinates[3]] = 1
121-
background_mask = binary_dilation(background_mask, iterations=self.dilation_strensegmentationh)
121+
background_mask = binary_dilation(background_mask, iterations=self.dilation_strength)
122122
background_mask = abs(
123123
background_mask.astype(np.float32) - dilated_object.astype(np.float32)
124124
) # casting booleans to do subtraction

0 commit comments

Comments
 (0)