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) 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): mask, im = segregation(im, 'hsv') mask = morpho(mask) mask, regions = sliding_window(mask) return regions, mask, im
def get_mask(self, im: np.array): # Color segmentation in HSV mask, im = segregation(im, 'hsv') mask = morpho(mask) mask, regions = integral(mask) return regions, 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