File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments