def get_mask(self, im: np.array):
"""
get_mask(im)
Function to compute the mask of an certain image,in this case is done
in RGB system
Parameters Value
----------------------
'im' Dataset image
Returns the mask, binary image with the detections.
"""
# Color segmentation in RGB
mask, im = segregation(im, 'rgb')
# Hole filling
mask = fill_holes(mask)
# Mask Blurring
mask = blur(mask)
# Compute the final mask
mask = discard_geometry.get_mask(mask)
return mask, im
def get_mask(self, im: np.array):
"""
get_mask(im)
Function to compute the mask of an certain image,in this case is done
in HSV system
Parameters Value
----------------------
'im' Dataset image
Returns the mask, binary image with the detections.
"""
# Color segmentation in HSV
mask, im = segregation(im, 'hsv')
# Mask morpho
mask = morpho(mask)
# Hole filling
mask = fill_holes(mask)
regions = get_cc_regions(mask)
# Compute the final mask
mask, regions = discard_geometry.get_mask(mask, regions)
return regions, mask, im
def get_mask(self, im: np.array):
"""
get_mask(im)
Function to compute the mask of an certain image,in this case is done
in HSV system to an image with Y-histogram equalised
Parameters Value
----------------------
'im' Dataset image
Returns the mask, binary image with the detections.
"""
# Equalization of the Y channel of the image
im = histogram_equalization(im, False)
# Color segmentation in HSV
mask, im = segregation(im, 'hsv')
# Mask Blurring
mask = blur(mask)
# Hole filling
mask = fill_holes(mask)
# Compute the final mask
mask = discard_geometry.get_mask(mask)
return mask, im
def get_mask(self, im: np.array):
# Color segmentation in HSV
mask, im = segregation(im, 'hsv')
mask = morpho(mask)
mask = fill_holes(mask)
mask, regions = convolution(mask)
mask, regions = discard_geometry.get_mask(mask, regions)
return regions, mask, im